Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
39 changes: 0 additions & 39 deletions src/gt4py/next/embedded/nd_array_field.py
Original file line number Diff line number Diff line change
Expand Up @@ -42,11 +42,6 @@
jax: Optional[ModuleType] = None # type: ignore[no-redef]
jnp: Optional[ModuleType] = None # type: ignore[no-redef]

try:
import dace
except ImportError:
dace: Optional[ModuleType] = None # type: ignore[no-redef]


def _get_nd_array_class(*fields: common.Field | core_defs.Scalar) -> type[NdArrayField]:
for f in fields:
Expand Down Expand Up @@ -468,40 +463,6 @@ def _slice(
assert common.is_relative_index_sequence(slice_)
return new_domain, slice_

if dace:

def _dace_data_ptr(self) -> int:
return self.__gt_buffer_info__.data_ptr

def _dace_descriptor(self) -> dace.data.Data:
return dace.data.create_datadescriptor(self.ndarray)

else:

def _dace_data_ptr(self) -> int:
raise NotImplementedError(
"data_ptr is only supported when the 'dace' module is available."
)

def _dace_descriptor(self) -> dace.data.Data:
raise NotImplementedError(
"__descriptor__ is only supported when the 'dace' module is available."
)

data_ptr = _dace_data_ptr
"""
Returns the pointer of the underlying data buffer.

Fully equivalent to `self.__gt_buffer_info__.data_ptr`. It is only defined to emulate the
PyTorch API for DaCe interoperability.

Note:
This method is experimental and will be likely removed in future versions.
"""

__descriptor__ = _dace_descriptor
"""Extension of NdArrayField adding SDFGConvertible support in GT4Py Programs."""


@dataclasses.dataclass(frozen=True)
class NdArrayConnectivityField(
Expand Down
11 changes: 1 addition & 10 deletions src/gt4py/next/ffront/decorator.py
Original file line number Diff line number Diff line change
Expand Up @@ -238,8 +238,7 @@ class Program(_CompilableGTEntryPointMixin[ffront_stages.DSLProgramDef]):
grid_type: The grid type (cartesian or unstructured) to be used. If not explicitly given
it will be deduced from actually occurring dimensions.
connectivities: A dictionary holding static/compile-time information about the offset providers.
For now, it is used for ahead of time compilation in DaCe orchestrated programs,
i.e. DaCe programs that call GT4Py Programs -SDFGConvertible interface-.
Used as the offset provider in `compile` when none is passed explicitly.
"""

@classmethod
Expand Down Expand Up @@ -416,14 +415,6 @@ def __call__(
self.definition_stage.definition(*args, **kwargs)


try:
from gt4py.next.program_processors.runners.dace.program import ( # type: ignore[assignment]
Program,
)
except ImportError:
pass


@dataclasses.dataclass(frozen=True)
class ProgramWithBoundArgs(Program):
bound_args: dict[str, float | int | bool] = dataclasses.field(default_factory=dict)
Expand Down
72 changes: 0 additions & 72 deletions src/gt4py/next/iterator/transforms/extractors.py

This file was deleted.

3 changes: 1 addition & 2 deletions src/gt4py/next/otf/options.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,8 +34,7 @@ class CompilationOptions:

# TODO(ricoh): replace with common.OffsetProviderType once the temporary pass doesn't require the runtime information
#: A dictionary holding static/compile-time information about the offset providers.
#: For now, it is used for ahead of time compilation in DaCe orchestrated programs,
#: i.e. DaCe programs that call GT4Py Programs -SDFGConvertible interface-.
#: Used as the offset provider in `compile` when none is passed explicitly.
connectivities: common.OffsetProvider | None = None

static_domains: bool = False
Expand Down
235 changes: 0 additions & 235 deletions src/gt4py/next/program_processors/runners/dace/program.py

This file was deleted.

Loading
Loading