Browse Source

!7243 fix special tbe op compile

Merge pull request !7243 from jjfeing/master
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
59fb711fcf
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/_extends/parallel_compile/tbe_compiler/compiler.py

+ 1
- 1
mindspore/_extends/parallel_compile/tbe_compiler/compiler.py View File

@@ -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:


Loading…
Cancel
Save