From a2555f99e3a234fb6709b0296ea238e6757a2d70 Mon Sep 17 00:00:00 2001 From: ksimpson Date: Tue, 10 Dec 2024 12:06:52 -0800 Subject: [PATCH] use Ralfs logic for ptx check --- cuda_core/tests/conftest.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cuda_core/tests/conftest.py b/cuda_core/tests/conftest.py index 0ff1f37b..30d80f6f 100644 --- 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