diff --git a/src/gt4py/cartesian/backend/__init__.py b/src/gt4py/cartesian/backend/__init__.py index 7a6f877295..e58c7a01a7 100644 --- a/src/gt4py/cartesian/backend/__init__.py +++ b/src/gt4py/cartesian/backend/__init__.py @@ -6,6 +6,8 @@ # Please, refer to the LICENSE file in the root directory. # SPDX-License-Identifier: BSD-3-Clause +from warnings import warn + from .base import ( REGISTRY, Backend, @@ -16,13 +18,6 @@ from_name, register, ) - - -try: - from .dace_backend import DaceCPUBackend, DaceGPUBackend -except ImportError: - pass - from .cuda_backend import CudaBackend from .gtcpp_backend import GTCpuIfirstBackend, GTCpuKfirstBackend, GTGpuBackend from .module_generator import BaseModuleGenerator @@ -47,5 +42,12 @@ ] -if "DaceCPUBackend" in globals(): +try: + from .dace_backend import DaceCPUBackend, DaceGPUBackend + __all__ += ["DaceCPUBackend", "DaceGPUBackend"] +except ImportError: + warn( + "GT4Py was unable to load DaCe. DaCe backends (`dace:cpu` and `dace:gpu`) will not be available.", + stacklevel=2, + )