diff --git a/cuda_core/cuda/core/_device.pyx b/cuda_core/cuda/core/_device.pyx index fee701b14e..1ea2df564c 100644 --- a/cuda_core/cuda/core/_device.pyx +++ b/cuda_core/cuda/core/_device.pyx @@ -1145,7 +1145,7 @@ class Device: from cuda.core._memory import DeviceMemoryResource self._memory_resource = DeviceMemoryResource(self._device_id) else: - from cuda.core._memory import _SynchronousMemoryResource + from cuda.core._memory._legacy import _SynchronousMemoryResource self._memory_resource = _SynchronousMemoryResource(self._device_id) return self._memory_resource diff --git a/cuda_core/cuda/core/_memory/_legacy.py b/cuda_core/cuda/core/_memory/_legacy.py index 919c37fc50..24ce88487e 100644 --- a/cuda_core/cuda/core/_memory/_legacy.py +++ b/cuda_core/cuda/core/_memory/_legacy.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 2024-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 2024-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # # SPDX-License-Identifier: Apache-2.0 @@ -17,7 +17,7 @@ driver, ) -__all__ = ["LegacyPinnedMemoryResource", "_SynchronousMemoryResource"] +__all__ = ["LegacyPinnedMemoryResource"] class LegacyPinnedMemoryResource(MemoryResource): diff --git a/cuda_core/tests/test_launcher.py b/cuda_core/tests/test_launcher.py index 899fdee4f5..aa427d6b3a 100644 --- a/cuda_core/tests/test_launcher.py +++ b/cuda_core/tests/test_launcher.py @@ -24,7 +24,7 @@ ProgramOptions, launch, ) -from cuda.core._memory import _SynchronousMemoryResource +from cuda.core._memory._legacy import _SynchronousMemoryResource from cuda.core._utils.cuda_utils import CUDAError