From 955ac912de0b32741403859c0b185f06f558082f Mon Sep 17 00:00:00 2001 From: Ziyue Jiang Date: Thu, 23 Jun 2022 10:32:42 +0800 Subject: [PATCH] remove log (#1160) --- colossalai/nn/graph/utils.py | 1 - 1 file changed, 1 deletion(-) diff --git a/colossalai/nn/graph/utils.py b/colossalai/nn/graph/utils.py index 9218bf994..40cc6320d 100644 --- a/colossalai/nn/graph/utils.py +++ b/colossalai/nn/graph/utils.py @@ -28,7 +28,6 @@ def register_colo_graph(input_pos: List[int], param_pos: List[int]) -> Callable: input_list.append(convert_to_colo_tensor(arg)) if isinstance(arg, torch.Tensor) and idx in param_pos: param_list.append(convert_to_colo_tensor(arg)) - print(f'Op {func}') # building the computing graph, inputs -> op if GraphGlobalEnv().graph_building: cur_op_node = GraphOpNode('linear', param_list)