Add some basic test coverage for cuda.core.experimental

[keenan-simpson]

Close#78.

This change adds a minimal testsuite to the repo so we can establish pipeline, and develop the tests alongside the cuda.core object model. It uses a pytest tree structure.

Example:

python -m pytest tests/
===================================================== test session starts ======================================================
platform linux -- Python 3.12.7, pytest-8.3.3, pluggy-1.5.0
benchmark: 4.0.0 (defaults: timer=time.perf_counter disable_gc=False min_rounds=5 min_time=0.000005 max_time=1.0 calibration_precision=10 warmup=False warmup_iterations=100000)
rootdir: /home/ksimpson/cuda-python/cuda-python/cuda_core
configfile: pyproject.toml
plugins: benchmark-4.0.0
collected 8 items
tests/test_context.py . [ 12%]
tests/test_device.py . [ 25%]
tests/test_event.py . [ 37%]
tests/test_launcher.py . [ 50%]
tests/test_memory.py . [ 62%]
tests/test_module.py . [ 75%]
tests/test_program.py . [ 87%]
tests/test_stream.py . [100%]
====================================================== 8 passed in 2.11s =======================================================

[copy-pr-bot]

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

[m3vaz]

/ok to test

[leofang]

Let's also hook up the code samples with the test system. We can do something like this: Collect the test files in-process, and loop over them.

[leofang]

@ksimpson-work heads up, we've done the move to the experimental submodule (#171), so you might need to merge with the latest main and make some changes 😅

[vzhurba01]

With this review I'm done looking at all the files. I'm sorry for all the nits, and thank you for handling the ones I've already shared! 🙏

[leofang]

Let's get the formatting feedbacks addressed quickly and move on. I've opened #195 to track an automated solution so that we don't need to spend time on reviewing formatting issues, which could sometimes be distracting/frustrating 🙂

[keenan-simpson]

Thankyou @vzhurba01 for your review. I have addressed all the comments and implemented all your nits as I agreed with all. Thanks @leofang for your review as well. I believe all comments have been addressed as of the most recent push

[vzhurba01]

LGTM

[leofang]

Thanks, Keenan/Vlad, for all the hard work! Unless I am missing something I don't think we have "relational tests" added yet (ex: Device.create_stream() returns a Stream; Stream.record() returns an Event, etc), but let's add them in the next PR.

[leofang]

@ksimpson-work do you see these warnings locally? This is concerning and making me wonder if we need to fix #141 as part of beta 1:

============================================================== warnings summary ===============================================================
<frozen importlib._bootstrap_external>:1297
<frozen importlib._bootstrap_external>:1297: DeprecationWarning: The cuda.cuda module is deprecated and will be removed in a future release, please switch to use the cuda.bindings.driver module instead.
<frozen importlib._bootstrap_external>:1297
<frozen importlib._bootstrap_external>:1297: DeprecationWarning: The cuda.cudart module is deprecated and will be removed in a future release, please switch to use the cuda.bindings.runtime module instead.
<frozen importlib._bootstrap_external>:1297
<frozen importlib._bootstrap_external>:1297: DeprecationWarning: The cuda.nvrtc module is deprecated and will be removed in a future release, please switch to use the cuda.bindings.nvrtc module instead.
tests/test_memory.py::test_buffer_initialization
tests/test_memory.py::test_buffer_copy_to
tests/test_memory.py::test_buffer_copy_from
/local/home/leof/miniforge3/envs/cuda_py_cuda126/lib/python3.12/site-packages/_pytest/unraisableexception.py:85: PytestUnraisableExceptionWarning: Exception ignored in: <function Buffer.__del__ at 0x7f3fb7ce1ee0>
Traceback (most recent call last):
File "/local/home/leof/dev/cuda-python/cuda_core/cuda/core/experimental/_memory.py", line 35, in __del__
self.close(default_stream())
File "/local/home/leof/dev/cuda-python/cuda_core/cuda/core/experimental/_memory.py", line 41, in close
self._mr.deallocate(self._ptr, self._size, stream)
File "/local/home/leof/dev/cuda-python/cuda_core/tests/test_memory.py", line 96, in deallocate
handle_return(cuda.cuMemFreeHost(ptr))
File "/local/home/leof/dev/cuda-python/cuda_core/cuda/core/experimental/_utils.py", line 57, in handle_return
_check_error(result[0], handle=handle)
File "/local/home/leof/dev/cuda-python/cuda_core/cuda/core/experimental/_utils.py", line 29, in _check_error
raise CUDAError(f"{name.decode()}: {desc.decode()}")
cuda.core.experimental._utils.CUDAError: CUDA_ERROR_INVALID_VALUE: invalid argument
warnings.warn(pytest.PytestUnraisableExceptionWarning(msg))
...

Could you check if explicitly calling Buffer.close() by the end of each test would make it go away? If so we should add it for now and circle back right after beta 1 (#141 needs a long-term solution).

[leofang]

Could you check if explicitly calling Buffer.close() by the end of each test would make it go away?

Confirmed locally that this is the way to go. mr.deallocate was meant to be called by buf.close, not by end users.

[leofang]

but let's add them in the next PR.

Tracked in #196. @vzhurba01 @ksimpson-work merge?

[leofang]

@ksimpson-work feel free to merge!