From 9ac5bf0e16b4e7163b8ffb4f6789df3bcf995f6c Mon Sep 17 00:00:00 2001 From: ksimpson Date: Mon, 25 Nov 2024 13:26:55 -0800 Subject: [PATCH] fix some test issues caused by bad merging on browser --- cuda_core/tests/test_memory.py | 4 ++-- cuda_core/tests/test_stream.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/cuda_core/tests/test_memory.py b/cuda_core/tests/test_memory.py index 0a70e52a..74c50b7a 100644 --- a/cuda_core/tests/test_memory.py +++ b/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: diff --git a/cuda_core/tests/test_stream.py b/cuda_core/tests/test_stream.py index e66b1684..faf0ad7d 100644 --- a/cuda_core/tests/test_stream.py +++ b/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)