Update base for Update on "[dtensor] PART 8: move tensor parallel api and tests to core distributed"

wanchaol · wanchaol · commit c3562a043e04 · 2022-11-15T06:19:51.000Z
This PR moves tensor/parallel folder and tests to torch.distributed. part of #88838 [ghstack-poisoned]
diff --git a/test/distributed/_tensor/test_dtensor_ops.py b/test/distributed/_tensor/test_dtensor_ops.py
@@ -332,6 +332,7 @@ def wrapped(fn):
     xfail("nn.functional.avg_pool2d"),
     xfail("nn.functional.avg_pool3d"),
     xfail("nn.functional.batch_norm"),
+    xfail("nn.functional.batch_norm", "without_cudnn"),
     xfail("nn.functional.bilinear"),
     xfail("nn.functional.binary_cross_entropy"),
     xfail("nn.functional.binary_cross_entropy_with_logits"),
