Skip to content

Commit

Permalink
[dace] Minor edit
Browse files Browse the repository at this point in the history
  • Loading branch information
edopao committed Oct 26, 2023
1 parent 289cd75 commit 61c538e
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -698,7 +698,7 @@ def _visit_shift(self, node: itir.FunCall) -> IteratorExpr:
expr = f"{internals[0]} + {internals[1]}"

shifted_value = self.add_expr_tasklet(
list(zip(args, internals)), expr, dace.dtypes.int64, "ind_addr"
list(zip(args, internals)), expr, dace.dtypes.int64, "shift"
)[0].value

shifted_index = {dim: value for dim, value in iterator.indices.items()}
Expand Down

0 comments on commit 61c538e

Please sign in to comment.