fixup! test(core.utils): assert key stability in cuLink driver-probe-failure test

Author: cpcloud

cuda_core/tests/test_program_cache.py

@@ -802,21 +802,23 @@ def _broken():
 
 def test_make_program_cache_key_driver_probe_failure_taints_ptx_under_cuLink(monkeypatch):
     """When the driver linker is active, _linker_backend_and_version
-    invokes _driver_version internally; a failing driver probe propagates
-    through the linker probe and must perturb the PTX key."""
+    invokes _driver_version internally; a failing driver probe must (a)
+    perturb the PTX key away from the success key, AND (b) be stable
+    across repeated calls so the persistent cache stays usable in the
+    failed environment."""
+    from cuda.core import _linker
     from cuda.core.utils import _program_cache
 
     def _broken():
         raise RuntimeError("driver probe failed")
 
-    # Force the cuLink driver path so _linker_backend_and_version reads driver_version.
-    from cuda.core import _linker
-
     monkeypatch.setattr(_linker, "_decide_nvjitlink_or_driver", lambda: True)
     k_ok = _make_key(code=".ptx", code_type="ptx")
     monkeypatch.setattr(_program_cache, "_driver_version", _broken)
-    k_broken = _make_key(code=".ptx", code_type="ptx")
-    assert k_ok != k_broken
+    k_broken1 = _make_key(code=".ptx", code_type="ptx")
+    k_broken2 = _make_key(code=".ptx", code_type="ptx")
+    assert k_ok != k_broken1
+    assert k_broken1 == k_broken2  # stable: same failure -> same key
 
 
 # ---------------------------------------------------------------------------