fixup! feat(core.utils): bytes-like code is NVVM-only

Author: cpcloud

cuda_core/cuda/core/utils/_program_cache.py

@@ -593,6 +593,14 @@ def make_program_cache_key(
     if isinstance(code, str):
         code_bytes = code.encode("utf-8")
     elif isinstance(code, (bytes, bytearray)):
+        # Program() only accepts bytes-like ``code`` for the NVVM backend
+        # (_program.pyx Program_init); c++/ptx require ``str``. Mirror that
+        # so the cache helper doesn't mint keys for inputs the real compile
+        # would reject.
+        if backend != "nvvm":
+            raise TypeError(
+                f"code must be str for code_type={code_type!r}; bytes/bytearray are only accepted for code_type='nvvm'."
+            )
         code_bytes = bytes(code)
     else:
         raise TypeError(f"code must be str or bytes, got {type(code).__name__}")

cuda_core/tests/test_program_cache.py

@@ -284,6 +284,20 @@ def test_make_program_cache_key_name_expressions_str_bytes_distinct():
     assert _make_key(name_expressions=("foo",)) != _make_key(name_expressions=(b"foo",))
 
 
+@pytest.mark.parametrize(
+    "code_type, target_type",
+    [
+        pytest.param("c++", "cubin", id="nvrtc"),
+        pytest.param("ptx", "cubin", id="ptx"),
+    ],
+)
+def test_make_program_cache_key_rejects_bytes_code_outside_nvvm(code_type, target_type):
+    """``Program()`` only accepts bytes-like code for NVVM; c++ and PTX
+    require str. The cache helper must mirror that rejection."""
+    with pytest.raises(TypeError, match="code must be str for code_type"):
+        _make_key(code=b"abc", code_type=code_type, target_type=target_type)
+
+
 @pytest.mark.parametrize(
     "code_type, code, target_type",
     [