Skip to content

Commit a1025c4

Browse files
authored
Skip JIT LTO example when PTX cannot load (#2151)
Signed-off-by: kiwigitops <kiwisclubco@gmail.com>
1 parent dad7ccb commit a1025c4

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

cuda_core/tests/example_tests/test_basic_examples.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
import pytest
1414

1515
from cuda.core import Device, ManagedMemoryResource, system
16+
from cuda.core._program import _can_load_generated_ptx
1617

1718
try:
1819
from cuda.bindings._test_helpers.pep723 import has_package_requirements_or_skip
@@ -82,6 +83,7 @@ def has_recent_memory_pool_support() -> bool:
8283
SYSTEM_REQUIREMENTS = {
8384
"memory_pool_resources.py": has_recent_memory_pool_support,
8485
"gl_interop_plasma.py": has_display,
86+
"jit_lto_fractal.py": _can_load_generated_ptx,
8587
"pytorch_example.py": lambda: (
8688
has_compute_capability_9_or_higher() and is_x86_64()
8789
), # PyTorch only provides CUDA support for x86_64

0 commit comments

Comments
 (0)