Trying to get more info

Author: mdboom

cuda_core/examples/saxpy.py

@@ -14,8 +14,15 @@
 # dependencies = ["cuda_bindings", "cuda_core", "cupy-cuda13x"]
 # ///
 
+
 import sys
 
+from cuda import pathfinder
+
+print(pathfinder.load_nvidia_dynamic_lib("nvrtc"))
+
+sys.exit(1)
+
 import cupy as cp
 
 from cuda.core import Device, LaunchConfig, Program, ProgramOptions, launch

cuda_core/tests/example_tests/test_basic_examples.py

@@ -53,6 +53,10 @@ def uv_installed() -> bool:
     return True
 
 
+def has_cuda_path() -> bool:
+    return os.environ.get("CUDA_PATH", os.environ.get("CUDA_HOME")) is not None
+
+
 PACKAGE_REQUIREMENTS = {
     "cuda_graphs.py": ["cupy"],
     "jit_lto_fractal.py": ["cupy"],
@@ -72,7 +76,8 @@ def uv_installed() -> bool:
     "pytorch_example.py": is_x86_64,  # PyTorch only provides CUDA support for x86_64
     "simple_multi_gpu_example.py": has_multiple_devices,
     "strided_memory_view_cpu.py": is_not_windows,
-    "thread_block_cluster.py": has_compute_capability_9_or_higher,
+    "thread_block_cluster.py": lambda: has_compute_capability_9_or_higher() and has_cuda_path(),
+    "tma_tensor_map.py": has_cuda_path,
 }