print is_fully_replicated in DebugString

PiperOrigin-RevId: 716860231
Intel-tensorflow · Jan 18, 2025 · e141532 · e141532
1 parent ee162a6
commit e141532
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/xla/python/ifrt/sharding.cc b/xla/python/ifrt/sharding.cc
@@ -743,9 +743,9 @@ std::string ConcreteEvenSharding::DebugString() const {
   DCHECK(this);
   return absl::StrFormat(
       "ConcreteEvenSharding(devices: %v, shape: %s, shard_shape: %s, "
-      "memory_kind: %v)",
-      *devices_, shape_.DebugString(), shard_shape_.DebugString(),
-      memory_kind_);
+      "memory_kind: %v, is_fully_replicated: %s)",
+      *devices_, shape_.DebugString(), shard_shape_.DebugString(), memory_kind_,
+      is_fully_replicated_ ? "true" : "false");
 }
 
 absl::StatusOr<std::unique_ptr<ShardingParamSharding>>