Skip to content

Commit d582448

Browse files
committed
Fix experimental.utils submodule access and test_device_id
[UNTESTED] Make utils accessible as cuda.core.experimental.utils submodule for backward compatibility. Also fix test_device_id to use cuda.core.system instead of cuda.core.experimental.system.
1 parent c414b7d commit d582448

2 files changed

Lines changed: 3 additions & 1 deletion

File tree

cuda_core/cuda/core/experimental/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -64,6 +64,8 @@ def _warn_deprecated():
6464
_warn_deprecated()
6565

6666
from cuda.core import utils # noqa: E402
67+
# Make utils accessible as a submodule for backward compatibility
68+
__import__("sys").modules[__spec__.name + ".utils"] = utils
6769
from cuda.core._device import Device # noqa: E402
6870
from cuda.core._event import Event, EventOptions # noqa: E402
6971
from cuda.core._graph import ( # noqa: E402

cuda_core/tests/test_device.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ def test_device_alloc(deinit_cuda):
4848

4949

5050
def test_device_id(deinit_cuda):
51-
for device in cuda.core.experimental.system.devices:
51+
for device in cuda.core.system.devices:
5252
device.set_current()
5353
assert device.device_id == handle_return(runtime.cudaGetDevice())
5454

0 commit comments

Comments
 (0)