Skip to content

Commit 371490e

Browse files
committed
fix test
1 parent c4a307a commit 371490e

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

cuda_core/tests/test_launcher.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -318,8 +318,10 @@ def test_launch_with_buffers_allocated_by_memory_resource(init_cuda, memory_reso
318318
mr = DeviceMemoryResource(dev.device_id)
319319
else:
320320
mr = _SynchronousMemoryResource(dev.device_id)
321+
name = memory_resource_class
321322
else:
322323
mr = memory_resource_class()
324+
name = str(mr)
323325

324326
# Allocate memory
325327
size = 1024
@@ -359,7 +361,7 @@ def test_launch_with_buffers_allocated_by_memory_resource(init_cuda, memory_reso
359361
stream.sync()
360362

361363
# Verify kernel operations
362-
assert cp.allclose(array, original * 3.0), f"{memory_resource_class.__name__} operation failed"
364+
assert cp.allclose(array, original * 3.0), f"{name} operation failed"
363365

364366
# Clean up
365367
buffer.close(stream)
@@ -368,4 +370,4 @@ def test_launch_with_buffers_allocated_by_memory_resource(init_cuda, memory_reso
368370
cp.cuda.Stream.null.use() # reset CuPy's current stream to the null stream
369371

370372
# Verify buffer is properly closed
371-
assert buffer.handle is None, f"{memory_resource_class.__name__} buffer should be closed"
373+
assert buffer.handle is None, f"{name} buffer should be closed"

0 commit comments

Comments
 (0)