Browse Source

!11243 set version when check supported

From: @jjfeing
Reviewed-by: @kisnwang,@zhoufeng54
Signed-off-by: @zhoufeng54
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 4 years ago
parent
commit
1fe7bea8de
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      mindspore/_extends/parallel_compile/tbe_compiler/helper.py

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

@@ -81,6 +81,7 @@ def _check_supported(kernel_info):
try:
op_name = kernel_info['op_info']['name']
is_dynamic_shape = kernel_info['op_info']['is_dynamic_shape']
te_set_version(kernel_info["op_info"]["socVersion"])
impl_path = build_in_impl_path
custom_flag = False
if 'impl_path' in kernel_info and kernel_info['impl_path'] is not None:


Loading…
Cancel
Save