Skip to content

Commit

Permalink
[dace] Fix previous commit
Browse files Browse the repository at this point in the history
  • Loading branch information
edopao committed Feb 21, 2024
1 parent 1ca6c48 commit 8c1f315
Showing 1 changed file with 3 additions and 5 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -259,10 +259,8 @@ def add_storage_for_temporaries(
):
# instead of using the actual size `end.value - begin.value` we allocate space for indexing from 0
# in order to avoid using dace array offsets
if begin.value != 0:
warnings.warn(
f"Array for temporary {tmp_name} allocated with size larger than the domain shape."
)
if not (isinstance(begin, SymbolExpr) and begin.value == "0"):
warnings.warn(f"Domain start offset for temporary {tmp_name} is ignored.")
tmp_symbols[str(shape_sym)] = end.value

return tmp_symbols
Expand Down Expand Up @@ -767,7 +765,7 @@ def _visit_parallel_stencil_closure(

def _visit_domain(
self, node: itir.FunCall, context: Context
) -> tuple[tuple[str, tuple[ValueExpr, ValueExpr]], ...]:
) -> tuple[tuple[str, tuple[SymbolExpr | ValueExpr, SymbolExpr | ValueExpr]], ...]:
assert isinstance(node.fun, itir.SymRef)
assert node.fun.id == "cartesian_domain" or node.fun.id == "unstructured_domain"

Expand Down

0 comments on commit 8c1f315

Please sign in to comment.