skip unhashable apis

PaddlePaddle · Dec 18, 2024 · d5cc8b8 · d5cc8b8
1 parent 3f99ca8
commit d5cc8b8
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/python/paddle/jit/sot/utils/paddle_api_config.py b/python/paddle/jit/sot/utils/paddle_api_config.py
@@ -132,6 +132,10 @@ def add_break_graph_apis(apis: list):
 
 
 def is_directly_run_api(api):
+    from .utils import hashable
+
+    if not hashable(api):
+        return False
     NATIVE_CODE_PURE_FUNCTIONS = {
         paddle.base.libpaddle.is_compiled_with_avx,
         paddle.base.libpaddle.is_compiled_with_cuda,