Skip to content

Commit c39d9a9

Browse files
remove windows stuff from async mempool until it is supported
1 parent 5fd7f46 commit c39d9a9

1 file changed

Lines changed: 8 additions & 0 deletions

File tree

cuda_core/tests/test_memory.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -227,6 +227,10 @@ def test_mempool():
227227
pytest.skip("Test requires CUDA 12 or higher")
228228
device = Device()
229229
device.set_current()
230+
231+
if not device.properties.memory_pools_supported:
232+
pytest.skip("Device does not support mempool operations")
233+
230234
pool_size = 2097152 # 2MB size
231235

232236
# Test basic pool creation
@@ -309,6 +313,10 @@ def test_mempool_properties(property_name, expected_type):
309313

310314
device = Device()
311315
device.set_current()
316+
317+
if not device.properties.memory_pools_supported:
318+
pytest.skip("Device does not support mempool operations")
319+
312320
pool_size = 2097152 # 2MB size
313321
mr = AsyncMempool.create(device.device_id, pool_size, ipc_enabled=False)
314322

0 commit comments

Comments
 (0)