Update debugging.py

pytorch · Nov 27, 2023 · 14e2bed · 14e2bed
1 parent eda1da0
commit 14e2bed
Showing 1 changed file with 2 additions and 10 deletions.
diff --git a/torch_xla/distributed/spmd/debugging.py b/torch_xla/distributed/spmd/debugging.py
@@ -175,16 +175,8 @@ def visualize_sharding(shape: torch.Size,
       top_padding, remainder = divmod(height - 2, 2)
       bottom_padding = top_padding + remainder
 
-      if use_color:
-        # color = _canonicalize_color(next(color_iter)[:3])
-        # text_color = _get_text_color(color)
-        # top_padding += 1
-        # bottom_padding += 1
-        # left_padding += 1
-        # right_padding += 1
-      else:
-        color = None
-        text_color = None
+      color = None
+      text_color = None
 
       padding = (top_padding, right_padding, bottom_padding, left_padding)
       padding = tuple(max(x, 0) for x in padding)