[dtensor] disable gpu tests in op db first

wanchaol · wanchaol · commit 0ff4cc8c9052 · 2023-01-19T07:32:42.000Z
There seems to be some issue with the cuda tests where our CI aren't capturing those failures (probably because of lacking 4 GPUs in CI environment). Disabling it first and debug later see #92343 [ghstack-poisoned]
diff --git a/test/distributed/_tensor/test_dtensor_ops.py b/test/distributed/_tensor/test_dtensor_ops.py
@@ -563,7 +563,9 @@ def wrapped(fn):
 
 
 OP_DB_WORLD_SIZE = 4
-DEVICE_TYPE = "cuda" if torch.cuda.is_available() and torch.cuda.device_count() >= OP_DB_WORLD_SIZE else "cpu"
+# DEVICE_TYPE = "cuda" if torch.cuda.is_available() and torch.cuda.device_count() >= OP_DB_WORLD_SIZE else "cpu"
+# TODO: debug cuda illegal memory access issue and re-enable cuda tests
+DEVICE_TYPE = "cpu"
 
 
 class TestDTensorOps(DTensorOpTestBase):
