apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-09-27 04:46:16 +08:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

fix import error (#2944)

Browse Source
This commit is contained in:
gaoziyuan
2025-07-22 14:06:01 +08:00
committed by GitHub
parent 8020927f50
commit 0eedbdaee0
1 changed files with 16 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
fastdeploy/import_ops.py
View File

@@ -70,6 +70,7 @@ def wrap_unified_op(original_cpp_ext_op, original_custom_op):
original_cpp_ext_op: Original C++ extension operator function.
original_custom_op: Original custom operator function.
"""
try:
@paddle.jit.marker.unified
@functools.wraps(original_custom_op)
@@ -84,6 +85,9 @@ def wrap_unified_op(original_cpp_ext_op, original_custom_op):
return res
return original_custom_op(*args, **kwargs)
except:
unified_op = None
logger.warning("Paddle version not support JIT mode.")
return unified_op