diff --git a/src/converter/pytorch_converter.py b/src/converter/pytorch_converter.py index 2371dccb..1dad1865 100644 --- 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]