renaming dtensor convolution unit tests

KingsleyLiu-NV · KingsleyLiu-NV · commit 9d57de823292 · 2023-11-19T21:33:26.000-08:00
diff --git a/test/distributed/_tensor/test_convolution_ops.py b/test/distributed/_tensor/test_convolution_ops.py
@@ -42,7 +42,7 @@ def world_size(self) -> int:
         return 2
 
     @with_comms
-    def test_downsampling_convolution_cuda(self):
+    def test_downsampling_convolution(self):
         device_mesh = DeviceMesh(self.device_type, list(range(self.world_size)))
         shard_spec = [Shard(3)]
 
@@ -91,7 +91,7 @@ def test_downsampling_convolution_cuda(self):
         self.assertTrue(bias_mse_rel <= 1e-6, f"Too large relative mse for bias tensor, expected less equal 1e-6, got {bias_mse_rel}")
 
     @with_comms
-    def test_depthwise_convolution_cuda(self):
+    def test_depthwise_convolution(self):
         device_mesh = DeviceMesh(self.device_type, list(range(self.world_size)))
         shard_spec = [Shard(3)]
 
