Skip to content

Commit

Permalink
refactor[next][daxce]: cleanup get_sdfg_args() parameter list (#1834)
Browse files Browse the repository at this point in the history
Remove unnecessary `kwargs` from parameter list of `get_sdfg_args()`.
  • Loading branch information
edopao authored Jan 29, 2025
1 parent ac47ca6 commit f732f08
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 4 deletions.
14 changes: 11 additions & 3 deletions src/gt4py/next/program_processors/runners/dace/sdfg_callable.py
Original file line number Diff line number Diff line change
Expand Up @@ -129,15 +129,23 @@ def get_sdfg_args(
*args: Any,
check_args: bool = False,
on_gpu: bool = False,
**kwargs: Any,
) -> dict[str, Any]:
"""Extracts the arguments needed to call the SDFG.
This function can handle the same arguments that are passed to dace runner.
This function can handle the arguments that are passed to the dace runner
and that end up in the decoration stage of the dace backend workflow.
Args:
sdfg: The SDFG for which we want to get the arguments.
offset_provider: Offset provider.
offset_provider: The offset provider.
args: The list of arguments passed to the dace runner.
check_args: If True, return only the arguments that are expected
according to the SDFG signature.
on_gpu: If True, this method ensures that the arrays for the
connectivity tables are allocated in GPU memory.
Returns:
A dictionary of keyword arguments to be passed in the SDFG call.
"""

dace_args = _get_args(sdfg, args)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -86,7 +86,6 @@ def decorated_program(
*flat_args,
check_args=False,
on_gpu=on_gpu,
use_field_canonical_representation=use_field_canonical_representation,
)

with dace.config.temporary_config():
Expand Down

0 comments on commit f732f08

Please sign in to comment.