Merge pull request #22269 from superbobry:main

PiperOrigin-RevId: 649395181

jax/_src/pjit.py

@@ -1147,7 +1147,7 @@ def unpack(key):
 
   # have we seen this function before at all?
   fun_name = getattr(f, '__qualname__', f)
-  if debug_info.func_src_info:
+  if debug_info is not None and debug_info.func_src_info:
     _, _, *rest = debug_info.func_src_info.split(' ')
     src_info = " defined at "  + ' '.join(rest)
   else:

tests/api_test.py

@@ -4618,6 +4618,11 @@ def f(key):
     for msg in cm.output:
       self.assertIn("TRACING CACHE MISS", msg)
 
+  def test_cache_miss_explanations_no_debug_info(self):
+    # ``operator.add`` is a built-in function and does not have debug info.
+    with config.explain_cache_misses(True):
+      jax.jit(operator.add)(42, 24)
+
   @parameterized.named_parameters([
       {"testcase_name": f"{dtype}", "dtype": dtype}
       for dtype in jtu.dtypes.custom_floats])