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

code format

pull/2364/head
oahzxl 2022-12-10 17:34:48 +08:00
parent d31e146687
commit 5de9e46381
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
chunk_codegen.py
View File

@ -49,7 +49,9 @@ def _is_non_compute_node(node):
def _is_non_compute_node_except_placeholder(node):
if (any(i in node.op for i in ["get_attr", "output"]) or any(i in node.name for i in ["getitem", "getattr"])):
if any(i in node.op for i in ["get_attr", "output"]) or any(
i in node.name for i in ["getitem", "getattr"]
):
return True
return False