2929)
3030from cuda .core .experimental ._dlpack import DLDeviceType
3131from cuda .core .experimental ._memory import IPCBufferDescriptor
32- from cuda .core .experimental ._utils .cuda_utils import handle_return
32+ from cuda .core .experimental ._utils .cuda_utils import CUDAError , handle_return
3333from cuda .core .experimental .utils import StridedMemoryView
3434from helpers import IS_WINDOWS
3535from helpers .buffers import DummyUnifiedMemoryResource
@@ -429,7 +429,13 @@ def test_vmm_allocator_policy_configuration():
429429 assert vmm_mr .config .granularity == "minimum"
430430
431431 # Test allocation with custom config
432- buffer = vmm_mr .allocate (8192 )
432+ try :
433+ buffer = vmm_mr .allocate (8192 )
434+ except CUDAError as exc :
435+ msg = str (exc )
436+ if "CUDA_ERROR_INVALID_DEVICE" in msg :
437+ pytest .xfail ("TODO(#1300): Failing on Jetson AGX Orin P3730" )
438+ raise
433439 assert buffer .size >= 8192
434440 assert buffer .device_id == device .device_id
435441
@@ -446,7 +452,13 @@ def test_vmm_allocator_policy_configuration():
446452 )
447453
448454 # Modify allocation policy
449- modified_buffer = vmm_mr .modify_allocation (buffer , 16384 , config = new_config )
455+ try :
456+ modified_buffer = vmm_mr .modify_allocation (buffer , 16384 , config = new_config )
457+ except CUDAError as exc :
458+ msg = str (exc )
459+ if "CUDA_ERROR_UNKNOWN" in msg :
460+ pytest .xfail ("TODO(#1300): Known to fail already with CTK 13.0 (Windows)" )
461+ raise
450462 assert modified_buffer .size >= 16384
451463 assert vmm_mr .config == new_config
452464 assert vmm_mr .config .self_access == "r"
0 commit comments