Skip to content

Commit 6a9c413

Browse files
authored
Merge branch 'main' into fix-fips-program-cache-hash
2 parents 404db9a + e481335 commit 6a9c413

1 file changed

Lines changed: 10 additions & 5 deletions

File tree

cuda_core/tests/test_memory.py

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1474,11 +1474,11 @@ def test_pinned_mr_numa_id_default_no_ipc(init_cuda):
14741474
device = Device()
14751475
skip_if_pinned_memory_unsupported(device)
14761476

1477-
mr = PinnedMemoryResource(PinnedMemoryResourceOptions())
1477+
mr = create_pinned_memory_resource_or_xfail(PinnedMemoryResourceOptions(), xfail_device=device)
14781478
assert mr.numa_id == -1
14791479
mr.close()
14801480

1481-
mr = PinnedMemoryResource(PinnedMemoryResourceOptions(ipc_enabled=False))
1481+
mr = create_pinned_memory_resource_or_xfail(PinnedMemoryResourceOptions(ipc_enabled=False), xfail_device=device)
14821482
assert mr.numa_id == -1
14831483
mr.close()
14841484

@@ -1497,7 +1497,9 @@ def test_pinned_mr_numa_id_default_with_ipc(init_cuda):
14971497
if expected_numa_id < 0:
14981498
pytest.skip("System does not support NUMA")
14991499

1500-
mr = PinnedMemoryResource(PinnedMemoryResourceOptions(ipc_enabled=True, max_size=POOL_SIZE))
1500+
mr = create_pinned_memory_resource_or_xfail(
1501+
PinnedMemoryResourceOptions(ipc_enabled=True, max_size=POOL_SIZE), xfail_device=device
1502+
)
15011503
assert mr.numa_id == expected_numa_id
15021504
mr.close()
15031505

@@ -1511,7 +1513,7 @@ def test_pinned_mr_numa_id_explicit(init_cuda):
15111513
if host_numa_id < 0:
15121514
pytest.skip("System does not support NUMA")
15131515

1514-
mr = PinnedMemoryResource(PinnedMemoryResourceOptions(numa_id=host_numa_id))
1516+
mr = create_pinned_memory_resource_or_xfail(PinnedMemoryResourceOptions(numa_id=host_numa_id), xfail_device=device)
15151517
assert mr.numa_id == host_numa_id
15161518
mr.close()
15171519

@@ -1520,7 +1522,10 @@ def test_pinned_mr_numa_id_explicit(init_cuda):
15201522
if not supports_ipc_mempool(device):
15211523
pytest.skip("Driver rejects IPC-enabled mempool creation on this platform")
15221524

1523-
mr = PinnedMemoryResource(PinnedMemoryResourceOptions(ipc_enabled=True, numa_id=host_numa_id, max_size=POOL_SIZE))
1525+
mr = create_pinned_memory_resource_or_xfail(
1526+
PinnedMemoryResourceOptions(ipc_enabled=True, numa_id=host_numa_id, max_size=POOL_SIZE),
1527+
xfail_device=device,
1528+
)
15241529
assert mr.numa_id == host_numa_id
15251530
mr.close()
15261531

0 commit comments

Comments
 (0)