repo-sync-2023-12-29T15:22:51+0800

anakinxc · Dec 29, 2023 · 3ae1ff1 · 3ae1ff1
1 parent 46e81fa
commit 3ae1ff1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/spu/utils/frontend.py b/spu/utils/frontend.py
@@ -88,7 +88,8 @@ def _argnames_partial_except(fn, static_argnames, kwargs):
     return functools.partial(fn, **static_kwargs), kwargs
 
 
-@cached(cache=LRUCache(maxsize=128), key=_jax_compilation_key)
+# FIXME: Figure out a proper way to hash lambda functions
+# @cached(cache=LRUCache(maxsize=128), key=_jax_compilation_key)
 def _jax_compilation(
     fn: Callable, static_argnums, static_argnames, args: List, kwargs: Dict
 ):