Update test_nn.py

eqy · web-flow · commit 4314872680be · 2024-12-31T13:35:52.000-08:00
diff --git a/test/test_nn.py b/test/test_nn.py
@@ -9961,7 +9961,7 @@ def test_upsamplingTrilinear3d(self, device, align_corners, memory_format):
             gradgradcheck(lambda x: F.interpolate(x, out_size, **kwargs), [input])
 
     @onlyCUDA
-    @skipCUDAIfRocm("launch bounds error out on ROCM")
+    @skipCUDAIfRocm(msg="launch bounds error out on ROCM")
     @dtypes(torch.half, torch.bfloat16)
     @largeTensorTest('40GB')
     def test_upsampling_64bit_indexing_channels_last(self, device, dtype):
