Merge pull request #171 from leofang/experimental

Move everything to `cuda.core.experimental`

Author: leofang

cuda_core/cuda/core/__init__.py

@@ -2,9 +2,4 @@
 #
 # SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
 
-from cuda.core._device import Device
-from cuda.core._event import EventOptions
-from cuda.core._launcher import LaunchConfig, launch
-from cuda.core._program import Program
-from cuda.core._stream import Stream, StreamOptions
 from cuda.core._version import __version__

cuda_core/cuda/core/experimental/__init__.pxd

@@ -0,0 +1,9 @@
+# Copyright (c) 2024, NVIDIA CORPORATION & AFFILIATES. ALL RIGHTS RESERVED.
+#
+# SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
+
+from cuda.core.experimental._device import Device
+from cuda.core.experimental._event import EventOptions
+from cuda.core.experimental._launcher import LaunchConfig, launch
+from cuda.core.experimental._program import Program
+from cuda.core.experimental._stream import Stream, StreamOptions

cuda_core/cuda/core/experimental/__init__.py

@@ -5,7 +5,7 @@
 from dataclasses import dataclass
 
 from cuda import cuda, cudart
-from cuda.core._utils import handle_return
+from cuda.core.experimental._utils import handle_return
 
 
 @dataclass

cuda_core/cuda/core/_context.py renamed to cuda_core/cuda/core/experimental/_context.py

@@ -7,11 +7,11 @@
 import warnings
 
 from cuda import cuda, cudart
-from cuda.core._utils import handle_return, ComputeCapability, CUDAError, \
+from cuda.core.experimental._utils import handle_return, ComputeCapability, CUDAError, \
                              precondition
-from cuda.core._context import Context, ContextOptions
-from cuda.core._memory import _DefaultAsyncMempool, Buffer, MemoryResource
-from cuda.core._stream import default_stream, Stream, StreamOptions
+from cuda.core.experimental._context import Context, ContextOptions
+from cuda.core.experimental._memory import _DefaultAsyncMempool, Buffer, MemoryResource
+from cuda.core.experimental._stream import default_stream, Stream, StreamOptions
 
 
 _tls = threading.local()
@@ -125,7 +125,7 @@ def set_current(self, ctx: Context=None) -> Union[Context, None]:
         Entry point of this object. Users always start a code by
         calling this method, e.g.
         
-        >>> from cuda.core import Device
+        >>> from cuda.core.experimental import Device
         >>> dev0 = Device(0)
         >>> dev0.set_current()
         >>> # ... do work on device 0 ...

cuda_core/cuda/core/_device.py renamed to cuda_core/cuda/core/experimental/_device.py

@@ -6,9 +6,9 @@
 from typing import Optional
 
 from cuda import cuda
-from cuda.core._utils import check_or_create_options
-from cuda.core._utils import CUDAError
-from cuda.core._utils import handle_return
+from cuda.core.experimental._utils import check_or_create_options
+from cuda.core.experimental._utils import CUDAError
+from cuda.core.experimental._utils import handle_return
 
 
 @dataclass

cuda_core/cuda/core/_dlpack.pxd renamed to cuda_core/cuda/core/experimental/_dlpack.pxd

@@ -15,7 +15,7 @@ import ctypes
 
 import numpy
 
-from cuda.core._memory import Buffer
+from cuda.core.experimental._memory import Buffer
 
 
 ctypedef cpp_complex.complex[float] cpp_single_complex

cuda_core/cuda/core/_dlpack.pyx renamed to cuda_core/cuda/core/experimental/_dlpack.pyx

@@ -8,11 +8,11 @@
 import numpy as np
 
 from cuda import cuda, cudart
-from cuda.core._kernel_arg_handler import ParamHolder
-from cuda.core._memory import Buffer
-from cuda.core._module import Kernel
-from cuda.core._stream import Stream
-from cuda.core._utils import CUDAError, check_or_create_options, handle_return
+from cuda.core.experimental._kernel_arg_handler import ParamHolder
+from cuda.core.experimental._memory import Buffer
+from cuda.core.experimental._module import Kernel
+from cuda.core.experimental._stream import Stream
+from cuda.core.experimental._utils import CUDAError, check_or_create_options, handle_return
 
 
 @dataclass