@@ -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