fix special tbe op compile

4 years ago · b9f97b60b0
parent 0d9fece038
commit b9f97b60b0
1 changed files with 1 additions and 1 deletions
--- a/mindspore/_extends/parallel_compile/tbe_compiler/compiler.py
+++ b/mindspore/_extends/parallel_compile/tbe_compiler/compiler.py
@ -107,7 +107,7 @@ def build_op(build_type, json_str):
            raise ValueError("Op:{} function {} is not supported by Tbe.".format(op_name, build_type))

        # call function
-        if kernel_name[0:19] == "bounding_box_encode":
+        if op_name == "bounding_box_encode":
            return op_func(*inputs_args, *outputs_args, *attrs_args, kernel_name_val=kernel_name)

        if is_dynamic_shape: