From 14e2bedd258c00c83007cc01d8bcca03f26688c9 Mon Sep 17 00:00:00 2001 From: Manfei <41607353+ManfeiBai@users.noreply.github.com> Date: Mon, 27 Nov 2023 15:19:29 -0800 Subject: [PATCH] Update debugging.py --- torch_xla/distributed/spmd/debugging.py | 12 ++---------- 1 file changed, 2 insertions(+), 10 deletions(-) diff --git a/torch_xla/distributed/spmd/debugging.py b/torch_xla/distributed/spmd/debugging.py index 44562d3e263..8d80929b5c7 100644 --- 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)