Skip to content

Commit

Permalink
Minor updates to SDFG
Browse files Browse the repository at this point in the history
  • Loading branch information
ThrudPrimrose committed Oct 24, 2024
1 parent d8141fd commit 96befbf
Showing 1 changed file with 4 additions and 8 deletions.
12 changes: 4 additions & 8 deletions dace/sdfg/sdfg.py
Original file line number Diff line number Diff line change
Expand Up @@ -1708,8 +1708,7 @@ def add_array(self,
total_size=None,
find_new_name=False,
alignment=0,
may_alias=False,
host_data=False) -> Tuple[str, dt.Array]:
may_alias=False) -> Tuple[str, dt.Array]:
""" Adds an array to the SDFG data descriptor store. """

# convert strings to int if possible
Expand Down Expand Up @@ -1737,8 +1736,7 @@ def add_array(self,
alignment=alignment,
debuginfo=debuginfo,
total_size=total_size,
may_alias=may_alias,
host_data=host_data)
may_alias=may_alias)

return self.add_datadesc(name, desc, find_new_name=find_new_name), desc

Expand Down Expand Up @@ -1877,8 +1875,7 @@ def add_scalar(self,
transient=False,
lifetime=dace.dtypes.AllocationLifetime.Scope,
debuginfo=None,
find_new_name=False,
host_data=False) -> Tuple[str, dt.Scalar]:
find_new_name=False) -> Tuple[str, dt.Scalar]:
""" Adds a scalar to the SDFG data descriptor store. """

if isinstance(dtype, type) and dtype in dtypes._CONSTANT_TYPES[:-1]:
Expand All @@ -1889,8 +1886,7 @@ def add_scalar(self,
storage=storage,
transient=transient,
lifetime=lifetime,
debuginfo=debuginfo,
host_data=host_data
debuginfo=debuginfo
)

return self.add_datadesc(name, desc, find_new_name=find_new_name), desc
Expand Down

0 comments on commit 96befbf

Please sign in to comment.