Skip to content

GitLab

Unverified Commit 955ac912 authored by Ziyue Jiang's avatar Ziyue Jiang Committed by GitHub
Browse files

remove log (#1160)

parent 4e67b2a8
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
colossalai/nn/graph/utils.py
View file @ 955ac912
...@@ -28,7 +28,6 @@ def register_colo_graph(input_pos: List[int], param_pos: List[int]) -> Callable: ...@@ -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)) input_list.append(convert_to_colo_tensor(arg))
if isinstance(arg, torch.Tensor) and idx in param_pos: if isinstance(arg, torch.Tensor) and idx in param_pos:
param_list.append(convert_to_colo_tensor(arg)) param_list.append(convert_to_colo_tensor(arg))
print(f'Op {func}')
# building the computing graph, inputs -> op # building the computing graph, inputs -> op
if GraphGlobalEnv().graph_building: if GraphGlobalEnv().graph_building:
cur_op_node = GraphOpNode('linear', param_list) cur_op_node = GraphOpNode('linear', param_list)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment