Update on "[dtensor] PART 6: move DTensor op tests to core distributed"

This PR moves DTensor op tests to core distributed, including prop_rule, pointwise op, matrix op tests, etc. [ghstack-poisoned]
pytorch · wanchaol · Nov 6, 2022 · Nov 6, 2022 · Nov 6, 2022 · Nov 6, 2022
commit 355e48ab6f9cd6d7019dc9895a20d78a0bf34492
diff --git a/test/distributed/_tensor/test_common_rules.py b/test/distributed/_tensor/test_common_rules.py
@@ -23,6 +23,12 @@ class CommonRulesTest(DTensorTestBase):
     def parse_schema(self, schema_str):
         return FunctionSchema.parse(schema_str)
 
+    @property
+    def world_size(self) -> int:
+        # hard code world size to 4 as we need to test
+        # at least with 2d mesh
+        return 4
+
     @with_comms
     def test_einop_basic_propagation(self):
         # plain einsum, mm