From 3d16af1c1bf1611bafc08f60ba1a517f9e42b909 Mon Sep 17 00:00:00 2001 From: jjfeing Date: Wed, 13 Jan 2021 21:23:44 +0800 Subject: [PATCH] set version in check support --- mindspore/_extends/parallel_compile/tbe_compiler/helper.py | 1 + 1 file changed, 1 insertion(+) diff --git a/mindspore/_extends/parallel_compile/tbe_compiler/helper.py b/mindspore/_extends/parallel_compile/tbe_compiler/helper.py index a74baa416c..d055c12e5b 100644 --- a/mindspore/_extends/parallel_compile/tbe_compiler/helper.py +++ b/mindspore/_extends/parallel_compile/tbe_compiler/helper.py @@ -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: