diff --git a/mindspore/_extends/parallel_compile/tbe_compiler/helper.py b/mindspore/_extends/parallel_compile/tbe_compiler/helper.py
index e223525e9a..a74baa416c 100644
--- a/mindspore/_extends/parallel_compile/tbe_compiler/helper.py
+++ b/mindspore/_extends/parallel_compile/tbe_compiler/helper.py
@@ -15,6 +15,7 @@
 """tbe process"""
 import sys
 import os
+from te.platform.cce_conf import te_set_version
 from .common import get_args, get_build_in_impl_path, TBEException
 
 build_in_impl_path = get_build_in_impl_path()
@@ -32,6 +33,7 @@ def _op_select_format(kernel_info):
     """
     try:
         op_name = kernel_info['op_info']['name']
+        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: