diff --git a/torch_xla/csrc/xla_sharding_util.h b/torch_xla/csrc/xla_sharding_util.h index 3e600be6871..697f320f575 100644 --- 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,