Merge branch 'main' into ksimpson/cuda_core_linker_155

Author: ksimpson-work

cuda_core/tests/test_memory.py

@@ -21,11 +21,11 @@ def __init__(self, device):
         self.device = device
 
     def allocate(self, size, stream=None) -> Buffer:
-        ptr = handle_return(cuda.cuMemAlloc(size))
+        ptr = handle_return(driver.cuMemAlloc(size))
         return Buffer(ptr=ptr, size=size, mr=self)
 
     def deallocate(self, ptr, size, stream=None):
-        handle_return(cuda.cuMemFree(ptr))
+        handle_return(driver.cuMemFree(ptr))
 
     @property
     def is_device_accessible(self) -> bool:

cuda_core/tests/test_stream.py

@@ -72,7 +72,7 @@ def test_stream_context(init_cuda):
     context = stream.context
     assert context is not None
 
-def test_stream_from_foreign_stream():
+def test_stream_from_foreign_stream(init_cuda):
     device = Device()
     other_stream = device.create_stream(options=StreamOptions())
     stream = device.create_stream(obj=other_stream)