Skip to content

Commit b4b2bf1

Browse files
committed
Add xfail in cuda_core/tests/test_memory.py: these tests are currently skipped in our CI; failures discovered elsewhere
1 parent 455889d commit b4b2bf1

1 file changed

Lines changed: 15 additions & 3 deletions

File tree

‎cuda_core/tests/test_memory.py‎

Lines changed: 15 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@
2929
)
3030
from cuda.core.experimental._dlpack import DLDeviceType
3131
from 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
3333
from cuda.core.experimental.utils import StridedMemoryView
3434
from helpers import IS_WINDOWS
3535
from 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

Comments
 (0)