From fb5f9a7b0ab6f385948b88f6b8ff0c852b8081d7 Mon Sep 17 00:00:00 2001 From: Pratyai Mazumder Date: Fri, 25 Oct 2024 14:48:50 +0200 Subject: [PATCH] Now the fix. The previously added test should pass now. --- dace/libraries/standard/nodes/reduce.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dace/libraries/standard/nodes/reduce.py b/dace/libraries/standard/nodes/reduce.py index fa231c07f2..970dfcef3a 100644 --- a/dace/libraries/standard/nodes/reduce.py +++ b/dace/libraries/standard/nodes/reduce.py @@ -103,7 +103,7 @@ def expansion(node: 'Reduce', state: SDFGState, sdfg: SDFG): 'reduce_init', {'_o%d' % i: '0:%s' % symstr(d) for i, d in enumerate(outedge.data.subset.size())}, {}, '__out = %s' % node.identity, - {'__out': dace.Memlet.simple('_out', ','.join(['_o%d' % i for i in range(output_dims)]))}, + {'__out': dace.Memlet.simple('_out', ','.join(['_o%d' % i for i in osqdim]))}, external_edges=True) else: nstate = nsdfg.add_state()