def _post_expand_trafos(sdfg: dace.SDFG): while inline_sdfgs(sdfg) or fuse_states(sdfg): pass sdfg.simplify() for node, _ in sdfg.all_nodes_recursive(): if isinstance(node, dace.nodes.MapEntry): node.collapse = len(node.range)
def _pre_expand_trafos(sdfg: dace.SDFG): sdfg.simplify()