From b9f97b60b0b118c13bb3dbaff1e5b961edba0aaf Mon Sep 17 00:00:00 2001 From: jjfeing Date: Tue, 13 Oct 2020 16:24:07 +0800 Subject: [PATCH] fix special tbe op compile --- mindspore/_extends/parallel_compile/tbe_compiler/compiler.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/_extends/parallel_compile/tbe_compiler/compiler.py b/mindspore/_extends/parallel_compile/tbe_compiler/compiler.py index e4b8397016..7bb4923bb5 100755 --- 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: