diff --git a/dace/transformation/interstate/sdfg_nesting.py b/dace/transformation/interstate/sdfg_nesting.py index 54986858bc..1c30a2f111 100644 --- a/dace/transformation/interstate/sdfg_nesting.py +++ b/dace/transformation/interstate/sdfg_nesting.py @@ -629,7 +629,7 @@ def _modify_access_to_access(self, matching_edge.data, use_dst_subset=True) new_memlet = in_memlet - new_memlet.other_subset = out_memlet.dst_subset + new_memlet.other_subset = out_memlet.subset inner_edge.data = new_memlet if len(nstate.out_edges(inner_edge.dst)) > 0: diff --git a/tests/python_frontend/structures/structure_python_test.py b/tests/python_frontend/structures/structure_python_test.py index 891c66ecac..d981d996f9 100644 --- a/tests/python_frontend/structures/structure_python_test.py +++ b/tests/python_frontend/structures/structure_python_test.py @@ -181,7 +181,7 @@ def rgf_leftToRight(A: BlockTriDiagonal, B: BlockTriDiagonal, n_: dace.int32, nb if __name__ == '__main__': - # test_read_structure() - # test_write_structure() - # test_local_structure() + test_read_structure() + test_write_structure() + test_local_structure() test_rgf()