Disable test_set functionalization ops test for now.

pytorch · Nov 13, 2023 · 794cbc7 · 794cbc7
1 parent 25f6dce
commit 794cbc7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_xla/csrc/xla_sharding_util.h b/torch_xla/csrc/xla_sharding_util.h
@@ -15,7 +15,7 @@ namespace torch_xla {
 
 class ShardingUtil {
  public:
-  // This maps to `torch_xla.experimental.xla_sharding.ShardingType` enum type.
+  // This maps to `torch_xla.distributed.spmd.ShardingType` enum type.
   enum ShardingType {
     REPLICATED = 0,
     MAXIMAL = 1,