Skip to content

Commit

Permalink
Fix parsing of tuple arguments on pragram call
Browse files Browse the repository at this point in the history
  • Loading branch information
edopao committed Sep 20, 2024
1 parent 0060d5d commit f01e394
Showing 1 changed file with 3 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
import dace
import numpy as np

from gt4py.next import common as gtx_common
from gt4py.next import common as gtx_common, utils as gtx_utils

from . import utility as dace_util

Expand Down Expand Up @@ -54,9 +54,10 @@ def _get_args(
sdfg: dace.SDFG, args: Sequence[Any], use_field_canonical_representation: bool
) -> dict[str, Any]:
sdfg_params: Sequence[str] = sdfg.arg_names
flat_args: Sequence[Any] = list(gtx_utils.flatten_nested_tuple(tuple(args)))
return {
sdfg_param: _convert_arg(arg, sdfg_param, use_field_canonical_representation)
for sdfg_param, arg in zip(sdfg_params, args, strict=True)
for sdfg_param, arg in zip(sdfg_params, flat_args, strict=True)
}


Expand Down

0 comments on commit f01e394

Please sign in to comment.