more debug

mlcommons · Oct 2, 2024 · 43f14c3 · 43f14c3
1 parent 9e0882b
commit 43f14c3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/converter/pytorch_converter.py b/src/converter/pytorch_converter.py
@@ -330,6 +330,7 @@ def get_protobuf_node_type_from_json_node(
         Returns:
             int: The corresponding Chakra node type.
         """
+        logging.debug(f"{json_node.is_gpu_op()=}, {json_node.name=}, {json_node_map[json_node.parent]=}, {json_node_map[json_node.parent].name=}")
         if json_node.is_gpu_op():
             if "ncclDevKernel_SendRecv" in json_node.name:
                 parent_node = json_node_map[json_node.parent]