github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

remove log (#1160)

pull/1161/head
Ziyue Jiang 2022-06-23 10:32:42 +08:00 committed by GitHub
parent 4e67b2a890
commit 955ac912de
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
colossalai/nn/graph/utils.py
View File

@ -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)