Skip to content

Commit d4569e9

Browse files
committed
test: handle deprecated calls
1 parent 6769cbf commit d4569e9

1 file changed

Lines changed: 8 additions & 4 deletions

File tree

cuda_core/tests/test_memory.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -365,7 +365,8 @@ def test_buffer_external_host():
365365

366366
@pytest.mark.parametrize("change_device", [True, False])
367367
def test_buffer_external_device(change_device):
368-
n = ccx_system.num_devices
368+
with pytest.deprecated_call():
369+
n = ccx_system.num_devices
369370
if n < 1:
370371
pytest.skip("No devices found")
371372
dev_id = n - 1
@@ -389,7 +390,8 @@ def test_buffer_external_device(change_device):
389390

390391
@pytest.mark.parametrize("change_device", [True, False])
391392
def test_buffer_external_pinned_alloc(change_device):
392-
n = ccx_system.num_devices
393+
with pytest.deprecated_call():
394+
n = ccx_system.num_devices
393395
if n < 1:
394396
pytest.skip("No devices found")
395397
dev_id = n - 1
@@ -414,7 +416,8 @@ def test_buffer_external_pinned_alloc(change_device):
414416

415417
@pytest.mark.parametrize("change_device", [True, False])
416418
def test_buffer_external_pinned_registered(change_device):
417-
n = ccx_system.num_devices
419+
with pytest.deprecated_call():
420+
n = ccx_system.num_devices
418421
if n < 1:
419422
pytest.skip("No devices found")
420423
dev_id = n - 1
@@ -447,7 +450,8 @@ def test_buffer_external_pinned_registered(change_device):
447450

448451
@pytest.mark.parametrize("change_device", [True, False])
449452
def test_buffer_external_managed(change_device):
450-
n = ccx_system.num_devices
453+
with pytest.deprecated_call():
454+
n = ccx_system.num_devices
451455
if n < 1:
452456
pytest.skip("No devices found")
453457
dev_id = n - 1

0 commit comments

Comments
 (0)