Update on "[dtensor] PART 7: move remaining DTensor tests to core distributed"

wanchaol · wanchaol · commit 79ce74f2c842 · 2022-11-10T23:39:10.000Z
This PR moves remaining tests, i.e. tensor_ops, op db tests to core distributed part of #88838 [ghstack-poisoned]
diff --git a/test/distributed/_tensor/test_dtensor_ops.py b/test/distributed/_tensor/test_dtensor_ops.py
@@ -30,7 +30,6 @@
     TEST_SKIPS,
     DTensorConverter,
     DEVICE_TYPE,
-    NUM_DEVICES,
 )
 
 # rewrite common size variables to sth can be sharded evenly
@@ -659,7 +658,7 @@ def check_dtensor_func(test_case, test_func, opinfo, dry_run=False):
 class TestDTensorOps(DTensorTestBase):
     @property
     def world_size(self) -> int:
-        return NUM_DEVICES
+        return 4
 
     # only allow float dytpe for now, we can relax this constraint
     # when feel necessary later (i.e when adding quantization support).
