Update on "[dtensor] support lt/gt op"

wanchaol · wanchaol · commit bfa6138dcee4 · 2023-10-05T10:41:03.000-07:00
This PR enables lt/gt aten op

[ghstack-poisoned]
diff --git a/torch/distributed/_tensor/dispatch.py b/torch/distributed/_tensor/dispatch.py
@@ -17,7 +17,7 @@
     OutputSharding,
     OutputSpecType,
 )
-from torch.distributed._tensor.placement_types import DTensorSpec, TensorMeta, Replicate
+from torch.distributed._tensor.placement_types import DTensorSpec, Replicate, TensorMeta
 from torch.distributed._tensor.random import is_rng_supported_mesh
 from torch.distributed._tensor.redistribute import redistribute_local_tensor
 from torch.distributed._tensor.sharding_prop import ShardingPropagator
