use Ralfs logic for ptx check

NVIDIA · Dec 10, 2024 · a2555f9 · a2555f9
1 parent d0d528b
commit a2555f9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cuda_core/tests/conftest.py b/cuda_core/tests/conftest.py
@@ -70,4 +70,4 @@ def clean_up_cffi_files():
 def can_load_generated_ptx():
     _, driver_ver = driver.cuDriverGetVersion()
     _, nvrtc_major, nvrtc_minor = nvrtc.nvrtcVersion()
-    return not (nvrtc_major * 1000 + nvrtc_minor * 10 > driver_ver)
+    return nvrtc_major * 1000 + nvrtc_minor * 10 <= driver_ver