Skip to content

Commit ccc8801

Browse files
pschuhGoogle-ML-Automation
authored andcommitted
Make c-api topology and PjRtClient versions produce identical platform_verison
strings to improve cache reuse between aot and actual runtime uses. Remove runtime_type as a cache key (compilation shouldn't depend on the runtime). PiperOrigin-RevId: 845452856
1 parent 9a391d7 commit ccc8801

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

jax/_src/cache_key.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -334,7 +334,6 @@ def _hash_serialized_compile_options(hash_obj, compile_options_obj,
334334
def _hash_platform(hash_obj, backend):
335335
_hash_string(hash_obj, backend.platform)
336336
_hash_string(hash_obj, backend.platform_version)
337-
_hash_string(hash_obj, backend.runtime_type)
338337

339338

340339
def _hash_xla_flags(hash_obj, extra_flag_prefixes: list[str]):

0 commit comments

Comments
 (0)