@@ -406,7 +406,7 @@ def test_make_program_cache_key_ptx_driver_ignored_fields_collapse(field, a, b,
406406 compiles differing only in these flags produce identical ObjectCode."""
407407 from cuda .core import _linker
408408
409- monkeypatch .setattr (_linker , "_decide_nvjitlink_or_driver " , lambda : True ) # driver
409+ monkeypatch .setattr (_linker , "_choose_backend " , lambda * _args , ** _kwargs : " driver" )
410410 k_a = _make_key (code = ".version 7.0" , code_type = "ptx" , options = _opts (** {field : a }))
411411 k_b = _make_key (code = ".version 7.0" , code_type = "ptx" , options = _opts (** {field : b }))
412412 assert k_a == k_b
@@ -430,7 +430,7 @@ def test_make_program_cache_key_ptx_ptxas_options_canonicalized(a, b, monkeypatc
430430 as equivalent so equivalent compiles don't miss the cache."""
431431 from cuda .core import _linker
432432
433- monkeypatch .setattr (_linker , "_decide_nvjitlink_or_driver " , lambda : False ) # nvJitLink
433+ monkeypatch .setattr (_linker , "_choose_backend " , lambda * _args , ** _kwargs : "nvjitlink" )
434434 k_a = _make_key (code = ".version 7.0" , code_type = "ptx" , options = _opts (ptxas_options = a ))
435435 k_b = _make_key (code = ".version 7.0" , code_type = "ptx" , options = _opts (ptxas_options = b ))
436436 assert k_a == k_b
@@ -440,7 +440,7 @@ def test_make_program_cache_key_ptx_driver_ignored_fields_still_matter_under_nvj
440440 """nvJitLink does honour those fields; they must still differentiate keys there."""
441441 from cuda .core import _linker
442442
443- monkeypatch .setattr (_linker , "_decide_nvjitlink_or_driver " , lambda : False ) # nvJitLink
443+ monkeypatch .setattr (_linker , "_choose_backend " , lambda * _args , ** _kwargs : "nvjitlink" )
444444 k_a = _make_key (code = ".version 7.0" , code_type = "ptx" , options = _opts (ftz = True ))
445445 k_b = _make_key (code = ".version 7.0" , code_type = "ptx" , options = _opts (ftz = False ))
446446 assert k_a != k_b
@@ -600,7 +600,7 @@ def test_make_program_cache_key_ptx_rejects_driver_linker_unsupported(option_kw,
600600 to exactly mirror the driver-linker's own gate."""
601601 from cuda .core import _linker
602602
603- monkeypatch .setattr (_linker , "_decide_nvjitlink_or_driver " , lambda : True ) # driver
603+ monkeypatch .setattr (_linker , "_choose_backend " , lambda * _args , ** _kwargs : " driver" )
604604 with pytest .raises (ValueError , match = "driver linker" ):
605605 _make_key (code = ".version 7.0" , code_type = "ptx" , options = _opts (** option_kw ))
606606
@@ -610,7 +610,7 @@ def test_make_program_cache_key_ptx_accepts_driver_linker_unsupported_with_nvjit
610610 rejected at key time."""
611611 from cuda .core import _linker
612612
613- monkeypatch .setattr (_linker , "_decide_nvjitlink_or_driver " , lambda : False ) # nvJitLink
613+ monkeypatch .setattr (_linker , "_choose_backend " , lambda * _args , ** _kwargs : "nvjitlink" )
614614 # Should not raise.
615615 _make_key (code = ".version 7.0" , code_type = "ptx" , options = _opts (time = True ))
616616
@@ -1104,7 +1104,7 @@ def test_make_program_cache_key_driver_probe_failure_taints_ptx_under_cuLink(mon
11041104 def _broken ():
11051105 raise RuntimeError ("driver probe failed" )
11061106
1107- monkeypatch .setattr (_linker , "_decide_nvjitlink_or_driver " , lambda : True )
1107+ monkeypatch .setattr (_linker , "_choose_backend " , lambda * _args , ** _kwargs : "driver" )
11081108 k_ok = _make_key (code = ".ptx" , code_type = "ptx" )
11091109 monkeypatch .setattr (_program_cache ._keys , "_driver_version" , _broken )
11101110 k_broken1 = _make_key (code = ".ptx" , code_type = "ptx" )
0 commit comments