Skip to content

GitLab

"tests/git@developer.sourcefind.cn:OpenDAS/colossalai.git" did not exist on "1beb85cc25f35d51083bde6fbaa99a5c4c7fd387"
Commit 9c5e028a authored by oahzxl's avatar oahzxl
Browse files

fix bug again

parent 7fd3b45a
Show whitespace changes
Inline Side-by-side
Showing with 19 additions and 16 deletions
chunk_codegen.py
View file @ 9c5e028a
...@@ -2003,16 +2003,19 @@ def emit_code_with_chunk( ...@@ -2003,16 +2003,19 @@ def emit_code_with_chunk(
) )
# ones like # ones like
if "ones_like" in node.name: if "ones_like" in node.name:
chunk_dim = chunk_search[region_idx]["node_chunk_dim"][ meta_node = chunk_region_search.index_tracer.node_list[node_idx]
chunk_region_search.index_tracer.node_list[node_idx] chunk_dim = chunk_search[region_idx]["node_chunk_dim"][meta_node][
]["chunk_dim"] "chunk_dim"
]
if _get_node_shape(meta_node)[chunk_dim] != 1:
source_node = meta_node.args[0].args[0]
if ( if (
_get_node_shape( source_node not in chunk_search[region_idx]["node_chunk_dim"]
chunk_region_search.index_tracer.node_list[node_idx] or chunk_search[region_idx]["node_chunk_dim"][source_node][
)[chunk_dim] "chunk_dim"
== 1 ]
is None
): ):
continue
chunk_slice = _gen_chunk_slice_dim( chunk_slice = _gen_chunk_slice_dim(
chunk_dim, "chunk_idx", _get_node_shape(node) chunk_dim, "chunk_idx", _get_node_shape(node)
) )
......
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