mindspore 2.4.1__cp310-cp310-manylinux1_x86_64.whl → 2.5.0__cp310-cp310-manylinux1_x86_64.whl

mindspore/lib/plugin/ascend/custom_ascendc_910/framework/npu_supported_ops.json

@@ -0,0 +1,10 @@
+{
+  "DecoderKvCache": {
+    "isGray": false,
+    "isHeavy": false
+  },
+  "PromptKvCache": {
+    "isGray": false,
+    "isHeavy": false
+  }
+}

mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/config/ascend910/aic-ascend910-ops-info.json

@@ -0,0 +1,182 @@
+{
+    "DecoderKvCache":{
+        "dynamicCompileStatic":{
+            "flag":"true"
+        },
+        "dynamicFormat":{
+            "flag":"true"
+        },
+        "dynamicRankSupport":{
+            "flag":"true"
+        },
+        "dynamicShapeSupport":{
+            "flag":"true"
+        },
+        "input0":{
+            "dtype":"float16,float32,int8,int16,int32,uint8,uint16,bfloat16",
+            "format":"ND,ND,ND,ND,ND,ND,ND,ND",
+            "name":"cache",
+            "paramType":"required",
+            "shape":"all",
+            "unknownshape_format":"ND,ND,ND,ND,ND,ND,ND,ND"
+        },
+        "input1":{
+            "dtype":"float16,float32,int8,int16,int32,uint8,uint16,bfloat16",
+            "format":"ND,ND,ND,ND,ND,ND,ND,ND",
+            "name":"update",
+            "paramType":"required",
+            "shape":"all",
+            "unknownshape_format":"ND,ND,ND,ND,ND,ND,ND,ND"
+        },
+        "input2":{
+            "dtype":"int64,int64,int64,int64,int64,int64,int64,int64",
+            "format":"ND,ND,ND,ND,ND,ND,ND,ND",
+            "name":"valid_seq_len",
+            "paramType":"required",
+            "shape":"all",
+            "unknownshape_format":"ND,ND,ND,ND,ND,ND,ND,ND"
+        },
+        "input3":{
+            "dtype":"int64,int64,int64,int64,int64,int64,int64,int64",
+            "format":"ND,ND,ND,ND,ND,ND,ND,ND",
+            "name":"batch_index",
+            "paramType":"required",
+            "shape":"all",
+            "unknownshape_format":"ND,ND,ND,ND,ND,ND,ND,ND"
+        },
+        "input4":{
+            "dtype":"int64,int64,int64,int64,int64,int64,int64,int64",
+            "format":"ND,ND,ND,ND,ND,ND,ND,ND",
+            "name":"seq_len_axis",
+            "paramType":"required",
+            "shape":"all",
+            "unknownshape_format":"ND,ND,ND,ND,ND,ND,ND,ND"
+        },
+        "input5":{
+            "dtype":"int64,int64,int64,int64,int64,int64,int64,int64",
+            "format":"ND,ND,ND,ND,ND,ND,ND,ND",
+            "name":"new_max_seq_len",
+            "paramType":"required",
+            "shape":"all",
+            "unknownshape_format":"ND,ND,ND,ND,ND,ND,ND,ND"
+        },
+        "input6":{
+            "dtype":"int64,int64,int64,int64,int64,int64,int64,int64",
+            "format":"ND,ND,ND,ND,ND,ND,ND,ND",
+            "name":"cur_max_seq_len",
+            "paramType":"required",
+            "shape":"all",
+            "unknownshape_format":"ND,ND,ND,ND,ND,ND,ND,ND"
+        },
+        "needCheckSupport":{
+            "flag":"false"
+        },
+        "opFile":{
+            "value":"decoder_kv_cache"
+        },
+        "opInterface":{
+            "value":"decoder_kv_cache"
+        },
+        "output0":{
+            "dtype":"float16,float32,int8,int16,int32,uint8,uint16,bfloat16",
+            "format":"ND,ND,ND,ND,ND,ND,ND,ND",
+            "name":"out",
+            "paramType":"required",
+            "shape":"all",
+            "unknownshape_format":"ND,ND,ND,ND,ND,ND,ND,ND"
+        },
+        "precision_reduce":{
+            "flag":"true"
+        }
+    },
+    "PromptKvCache":{
+        "dynamicCompileStatic":{
+            "flag":"true"
+        },
+        "dynamicFormat":{
+            "flag":"true"
+        },
+        "dynamicRankSupport":{
+            "flag":"true"
+        },
+        "dynamicShapeSupport":{
+            "flag":"true"
+        },
+        "input0":{
+            "dtype":"float16,float32,int8,int16,int32,uint8,uint16,bfloat16",
+            "format":"ND,ND,ND,ND,ND,ND,ND,ND",
+            "name":"cache",
+            "paramType":"required",
+            "shape":"all",
+            "unknownshape_format":"ND,ND,ND,ND,ND,ND,ND,ND"
+        },
+        "input1":{
+            "dtype":"float16,float32,int8,int16,int32,uint8,uint16,bfloat16",
+            "format":"ND,ND,ND,ND,ND,ND,ND,ND",
+            "name":"update",
+            "paramType":"required",
+            "shape":"all",
+            "unknownshape_format":"ND,ND,ND,ND,ND,ND,ND,ND"
+        },
+        "input2":{
+            "dtype":"int64,int64,int64,int64,int64,int64,int64,int64",
+            "format":"ND,ND,ND,ND,ND,ND,ND,ND",
+            "name":"valid_seq_len",
+            "paramType":"required",
+            "shape":"all",
+            "unknownshape_format":"ND,ND,ND,ND,ND,ND,ND,ND"
+        },
+        "input3":{
+            "dtype":"int64,int64,int64,int64,int64,int64,int64,int64",
+            "format":"ND,ND,ND,ND,ND,ND,ND,ND",
+            "name":"batch_index",
+            "paramType":"required",
+            "shape":"all",
+            "unknownshape_format":"ND,ND,ND,ND,ND,ND,ND,ND"
+        },
+        "input4":{
+            "dtype":"int64,int64,int64,int64,int64,int64,int64,int64",
+            "format":"ND,ND,ND,ND,ND,ND,ND,ND",
+            "name":"seq_len_axis",
+            "paramType":"required",
+            "shape":"all",
+            "unknownshape_format":"ND,ND,ND,ND,ND,ND,ND,ND"
+        },
+        "input5":{
+            "dtype":"int64,int64,int64,int64,int64,int64,int64,int64",
+            "format":"ND,ND,ND,ND,ND,ND,ND,ND",
+            "name":"new_max_seq_len",
+            "paramType":"required",
+            "shape":"all",
+            "unknownshape_format":"ND,ND,ND,ND,ND,ND,ND,ND"
+        },
+        "input6":{
+            "dtype":"int64,int64,int64,int64,int64,int64,int64,int64",
+            "format":"ND,ND,ND,ND,ND,ND,ND,ND",
+            "name":"cur_max_seq_len",
+            "paramType":"required",
+            "shape":"all",
+            "unknownshape_format":"ND,ND,ND,ND,ND,ND,ND,ND"
+        },
+        "needCheckSupport":{
+            "flag":"false"
+        },
+        "opFile":{
+            "value":"prompt_kv_cache"
+        },
+        "opInterface":{
+            "value":"prompt_kv_cache"
+        },
+        "output0":{
+            "dtype":"float16,float32,int8,int16,int32,uint8,uint16,bfloat16",
+            "format":"ND,ND,ND,ND,ND,ND,ND,ND",
+            "name":"out",
+            "paramType":"required",
+            "shape":"all",
+            "unknownshape_format":"ND,ND,ND,ND,ND,ND,ND,ND"
+        },
+        "precision_reduce":{
+            "flag":"true"
+        }
+    }
+}

mindspore/lib/plugin/ascend/{custom_ascendc_ops/op_impl/ai_core/tbe/custom_ascendc_ops_impl → custom_ascendc_910/op_impl/ai_core/tbe/custom_ascendc_910_impl}/dynamic/decoder_kv_cache.py

@@ -1,3 +1,8 @@
+#!/usr/bin/env python
+# -*- coding: UTF-8 -*-
+"""
+Copyright (c) Huawei Technologies Co., Ltd. 2024-2025. All rights reserved.
+"""
 
 import os, sys
 import ctypes
@@ -6,6 +11,7 @@ import shutil
 from tbe.common.platform import get_soc_spec
 from tbe.common.utils import para_check
 from tbe.tikcpp import compile_op, replay_op, check_op_cap, generalize_op_params, get_code_channel, OpInfo
+from tbe.tikcpp.compile_op import CommonUtility, AscendCLogLevel
 from tbe.common.buildcfg import get_default_build_config
 from impl.util.platform_adapter import tbe_register
 from tbe.common.buildcfg import get_current_build_config
@@ -28,6 +34,7 @@ DTYPE_MAP = {"float32": ["DT_FLOAT", "float"],
     "dual_sub_int8": ["DT_DUAL_SUB_INT8", "unknown"],
     "dual_sub_uint8": ["DT_DUAL_SUB_UINT8", "unknown"],
     "string": ["DT_STRING", "unknown"],
+    "complex32": ["DT_COMPLEX32", "unknown"],
     "complex64": ["DT_COMPLEX64", "unknown"],
     "complex128": ["DT_COMPLEX128", "unknown"],
     "qint8": ["DT_QINT8", "unknown"],
@@ -49,21 +56,23 @@ def add_dtype_fmt_option_single(x, x_n, is_ref: bool = False):
     options.append("-DORIG_DTYPE_{n}={ot}".format(n=x_n_in_kernel, ot=DTYPE_MAP.get(x_dtype)[0]))
     options.append("-DFORMAT_{n}=FORMAT_{f}".format(n=x_n_in_kernel, f=x_fmt))
     return options
- 
+
 def get_dtype_fmt_options(__inputs__, __outputs__):
     options = []
+    input_names = ['cache', 'update', 'valid_seq_len', 'batch_index', 'seq_len_axis', 'new_max_seq_len', 'cur_max_seq_len']
+    output_names = ['out']
     unique_param_name_set = set()
-    for x in __inputs__:
+    for idx, x in enumerate(__inputs__):
         if x is None:
             continue
-        x_n = x.get("param_name")[:-5].upper()
+        x_n = input_names[idx].upper()
         unique_param_name_set.add(x_n)
         options += add_dtype_fmt_option_single(x, x_n)
- 
-    for x in __outputs__:
+
+    for idx, x in enumerate(__outputs__):
         if x is None:
             continue
-        x_n = x.get("param_name")[:-5].upper()
+        x_n = output_names[idx].upper()
         if x_n in unique_param_name_set:
             options += add_dtype_fmt_option_single(x, x_n, True)
         else:
@@ -74,27 +83,40 @@ def load_dso(so_path):
     try:
         ctypes.CDLL(so_path)
     except OSError as error :
-        print(error)
+        CommonUtility.print_compile_log("", error, AscendCLogLevel.LOG_ERROR)
         raise RuntimeError("cannot open %s" %(so_path))
     else:
-        print("load so succ ", so_path)
+        msg = "load so succ " + so_path
+        CommonUtility.print_compile_log("", msg, AscendCLogLevel.LOG_INFO)
 
 def get_shortsoc_compile_option(compile_option_list: list, shortsoc:str):
     compile_options = []
     if shortsoc in compile_option_list:
-        compile_options = compile_option_list[shortsoc]
-    elif '__ALLSOC__' in compile_option_list:
-        compile_options = compile_option_list['__ALLSOC__']
+        compile_options.extend(compile_option_list[shortsoc])
+    if '__ALLSOC__' in compile_option_list:
+        compile_options.extend(compile_option_list['__ALLSOC__'])
     return compile_options
 
 def get_kernel_source(src_file, dir_snake, dir_ex):
     src_ex = os.path.join(PYF_PATH, "..", "ascendc", dir_ex, src_file)
     if os.path.exists(src_ex):
         return src_ex
+    src = os.environ.get('BUILD_KERNEL_SRC')
+    if src and os.path.exists(src):
+        return src
     src = os.path.join(PYF_PATH, "..", "ascendc", dir_snake, src_file)
     if os.path.exists(src):
         return src
     src = os.path.join(PYF_PATH, src_file)
+    if os.path.exists(src):
+        return src
+    src = os.path.join(PYF_PATH, "..", "ascendc", dir_snake, dir_snake + ".cpp")
+    if os.path.exists(src):
+        return src
+    src = os.path.join(PYF_PATH, "..", "ascendc", dir_ex, dir_ex + ".cpp")
+    if os.path.exists(src):
+        return src
+    src = os.path.join(PYF_PATH, "..", "ascendc", os.path.splitext(src_file)[0], src_file)
     if os.path.exists(src):
         return src
     return src_ex
@@ -109,6 +131,8 @@ def _build_args(cache_in__, update_in__, valid_seq_len_in__, batch_index_in__, s
                 __inputs__.append(arg[0])
             else:
                 __inputs__.append(arg)
+        else:
+            __inputs__.append(arg)
     __outputs__ = []
     for arg in [out_out_]:
         if arg != None:
@@ -118,24 +142,30 @@ def _build_args(cache_in__, update_in__, valid_seq_len_in__, batch_index_in__, s
                 __outputs__.append(arg[0])
             else:
                 __outputs__.append(arg)
+        else:
+            __outputs__.append(arg)
     __attrs__ = []
     return __inputs__, __outputs__, __attrs__
 
 @tbe_register.register_operator("DecoderKvCache", trans_bool_to_s8=False)
 @para_check.check_op_params(para_check.REQUIRED_INPUT, para_check.REQUIRED_INPUT, para_check.REQUIRED_INPUT, para_check.REQUIRED_INPUT, para_check.REQUIRED_INPUT, para_check.REQUIRED_INPUT, para_check.REQUIRED_INPUT, para_check.REQUIRED_OUTPUT, para_check.KERNEL_NAME)
 def decoder_kv_cache(cache_in__, update_in__, valid_seq_len_in__, batch_index_in__, seq_len_axis_in__, new_max_seq_len_in__, cur_max_seq_len_in__, out_out_, kernel_name="decoder_kv_cache", impl_mode=""):
+    # do ascendc build step
     if get_current_build_config("enable_op_prebuild"):
         return
     __inputs__, __outputs__, __attrs__ = _build_args(cache_in__, update_in__, valid_seq_len_in__, batch_index_in__, seq_len_axis_in__, new_max_seq_len_in__, cur_max_seq_len_in__, out_out_)
     options = get_dtype_fmt_options(__inputs__, __outputs__)
     options += ["-x", "cce"]
-    bisheng = shutil.which("bisheng")
+    bisheng = os.environ.get('BISHENG_REAL_PATH')
+    if bisheng is None:
+        bisheng = shutil.which("bisheng")
     if bisheng != None:
         bisheng_path = os.path.dirname(bisheng)
         tikcpp_path = os.path.realpath(os.path.join(bisheng_path, "..", "..", "tikcpp"))
     else:
         tikcpp_path = os.path.realpath("/usr/local/Ascend/latest/compiler/tikcpp")
     options.append("-I" + tikcpp_path)
+    options.append("-I" + os.path.join(tikcpp_path, "..", "..", "include"))
     options.append("-I" + os.path.join(tikcpp_path, "tikcfw"))
     options.append("-I" + os.path.join(tikcpp_path, "tikcfw", "impl"))
     options.append("-I" + os.path.join(tikcpp_path, "tikcfw", "interface"))
@@ -144,8 +174,10 @@ def decoder_kv_cache(cache_in__, update_in__, valid_seq_len_in__, batch_index_in
         options.append("-DHIGH_PERFORMANCE=1")
     elif impl_mode == "high_precision":
         options.append("-DHIGH_PRECISION=1")
-    if get_default_build_config("enable_deterministic_mode") == 1:
-        options.append("-DDETEMINISTIC_MODE=1")
+    if get_current_build_config("enable_deterministic_mode") == 1:
+        options.append("-DDETERMINISTIC_MODE=1")
+    else:
+        options.append("-DDETERMINISTIC_MODE=0")
 
     custom_compile_options = {},
     custom_all_compile_options = {},
@@ -162,11 +194,14 @@ def decoder_kv_cache(cache_in__, update_in__, valid_seq_len_in__, batch_index_in
     ascendc_src_file = "decoder_kv_cache.cpp"
     src = get_kernel_source(ascendc_src_file, ascendc_src_dir, ascendc_src_dir_ex)
 
-    print("start compile Ascend C operator DecoderKvCache. kernel name is " + kernel_name)
+    msg = "start compile Acend C Operator DecoderKvCache, kernel name is " + kernel_name
+    CommonUtility.print_compile_log("", msg, AscendCLogLevel.LOG_INFO)
     op_type = "DecoderKvCache"
     code_channel = get_code_channel(src, kernel_name, op_type, options)
     op_info = OpInfo(kernel_name = kernel_name, op_type = op_type, inputs = __inputs__, outputs = __outputs__,\
-        attrs = __attrs__, impl_mode = impl_mode, origin_inputs=[cache_in__, update_in__, valid_seq_len_in__, batch_index_in__, seq_len_axis_in__, new_max_seq_len_in__, cur_max_seq_len_in__], origin_outputs = [out_out_])
+        attrs = __attrs__, impl_mode = impl_mode, origin_inputs=[cache_in__, update_in__, valid_seq_len_in__, batch_index_in__, seq_len_axis_in__, new_max_seq_len_in__, cur_max_seq_len_in__], origin_outputs = [out_out_],\
+                param_type_dynamic = False, mc2_ctx = [], param_type_list = ['required', 'required', 'required', 'required', 'required', 'required', 'required', 'required'], init_value_list = [None],\
+                output_shape_depend_on_compute = [])
     compile_op(src, origin_func_name, op_info, options, code_channel, '{}')
 
 def op_select_format(cache_in__, update_in__, valid_seq_len_in__, batch_index_in__, seq_len_axis_in__, new_max_seq_len_in__, cur_max_seq_len_in__, out_out_, impl_mode=""):

mindspore/lib/plugin/ascend/{custom_ascendc_ops/op_impl/ai_core/tbe/custom_ascendc_ops_impl → custom_ascendc_910/op_impl/ai_core/tbe/custom_ascendc_910_impl}/dynamic/prompt_kv_cache.py

@@ -1,3 +1,8 @@
+#!/usr/bin/env python
+# -*- coding: UTF-8 -*-
+"""
+Copyright (c) Huawei Technologies Co., Ltd. 2024-2025. All rights reserved.
+"""
 
 import os, sys
 import ctypes
@@ -6,6 +11,7 @@ import shutil
 from tbe.common.platform import get_soc_spec
 from tbe.common.utils import para_check
 from tbe.tikcpp import compile_op, replay_op, check_op_cap, generalize_op_params, get_code_channel, OpInfo
+from tbe.tikcpp.compile_op import CommonUtility, AscendCLogLevel
 from tbe.common.buildcfg import get_default_build_config
 from impl.util.platform_adapter import tbe_register
 from tbe.common.buildcfg import get_current_build_config
@@ -28,6 +34,7 @@ DTYPE_MAP = {"float32": ["DT_FLOAT", "float"],
     "dual_sub_int8": ["DT_DUAL_SUB_INT8", "unknown"],
     "dual_sub_uint8": ["DT_DUAL_SUB_UINT8", "unknown"],
     "string": ["DT_STRING", "unknown"],
+    "complex32": ["DT_COMPLEX32", "unknown"],
     "complex64": ["DT_COMPLEX64", "unknown"],
     "complex128": ["DT_COMPLEX128", "unknown"],
     "qint8": ["DT_QINT8", "unknown"],
@@ -49,21 +56,23 @@ def add_dtype_fmt_option_single(x, x_n, is_ref: bool = False):
     options.append("-DORIG_DTYPE_{n}={ot}".format(n=x_n_in_kernel, ot=DTYPE_MAP.get(x_dtype)[0]))
     options.append("-DFORMAT_{n}=FORMAT_{f}".format(n=x_n_in_kernel, f=x_fmt))
     return options
- 
+
 def get_dtype_fmt_options(__inputs__, __outputs__):
     options = []
+    input_names = ['cache', 'update', 'valid_seq_len', 'batch_index', 'seq_len_axis', 'new_max_seq_len', 'cur_max_seq_len']
+    output_names = ['out']
     unique_param_name_set = set()
-    for x in __inputs__:
+    for idx, x in enumerate(__inputs__):
         if x is None:
             continue
-        x_n = x.get("param_name")[:-5].upper()
+        x_n = input_names[idx].upper()
         unique_param_name_set.add(x_n)
         options += add_dtype_fmt_option_single(x, x_n)
- 
-    for x in __outputs__:
+
+    for idx, x in enumerate(__outputs__):
         if x is None:
             continue
-        x_n = x.get("param_name")[:-5].upper()
+        x_n = output_names[idx].upper()
         if x_n in unique_param_name_set:
             options += add_dtype_fmt_option_single(x, x_n, True)
         else:
@@ -74,27 +83,40 @@ def load_dso(so_path):
     try:
         ctypes.CDLL(so_path)
     except OSError as error :
-        print(error)
+        CommonUtility.print_compile_log("", error, AscendCLogLevel.LOG_ERROR)
         raise RuntimeError("cannot open %s" %(so_path))
     else:
-        print("load so succ ", so_path)
+        msg = "load so succ " + so_path
+        CommonUtility.print_compile_log("", msg, AscendCLogLevel.LOG_INFO)
 
 def get_shortsoc_compile_option(compile_option_list: list, shortsoc:str):
     compile_options = []
     if shortsoc in compile_option_list:
-        compile_options = compile_option_list[shortsoc]
-    elif '__ALLSOC__' in compile_option_list:
-        compile_options = compile_option_list['__ALLSOC__']
+        compile_options.extend(compile_option_list[shortsoc])
+    if '__ALLSOC__' in compile_option_list:
+        compile_options.extend(compile_option_list['__ALLSOC__'])
     return compile_options
 
 def get_kernel_source(src_file, dir_snake, dir_ex):
     src_ex = os.path.join(PYF_PATH, "..", "ascendc", dir_ex, src_file)
     if os.path.exists(src_ex):
         return src_ex
+    src = os.environ.get('BUILD_KERNEL_SRC')
+    if src and os.path.exists(src):
+        return src
     src = os.path.join(PYF_PATH, "..", "ascendc", dir_snake, src_file)
     if os.path.exists(src):
         return src
     src = os.path.join(PYF_PATH, src_file)
+    if os.path.exists(src):
+        return src
+    src = os.path.join(PYF_PATH, "..", "ascendc", dir_snake, dir_snake + ".cpp")
+    if os.path.exists(src):
+        return src
+    src = os.path.join(PYF_PATH, "..", "ascendc", dir_ex, dir_ex + ".cpp")
+    if os.path.exists(src):
+        return src
+    src = os.path.join(PYF_PATH, "..", "ascendc", os.path.splitext(src_file)[0], src_file)
     if os.path.exists(src):
         return src
     return src_ex
@@ -109,6 +131,8 @@ def _build_args(cache_in__, update_in__, valid_seq_len_in__, batch_index_in__, s
                 __inputs__.append(arg[0])
             else:
                 __inputs__.append(arg)
+        else:
+            __inputs__.append(arg)
     __outputs__ = []
     for arg in [out_out_]:
         if arg != None:
@@ -118,24 +142,30 @@ def _build_args(cache_in__, update_in__, valid_seq_len_in__, batch_index_in__, s
                 __outputs__.append(arg[0])
             else:
                 __outputs__.append(arg)
+        else:
+            __outputs__.append(arg)
     __attrs__ = []
     return __inputs__, __outputs__, __attrs__
 
 @tbe_register.register_operator("PromptKvCache", trans_bool_to_s8=False)
 @para_check.check_op_params(para_check.REQUIRED_INPUT, para_check.REQUIRED_INPUT, para_check.REQUIRED_INPUT, para_check.REQUIRED_INPUT, para_check.REQUIRED_INPUT, para_check.REQUIRED_INPUT, para_check.REQUIRED_INPUT, para_check.REQUIRED_OUTPUT, para_check.KERNEL_NAME)
 def prompt_kv_cache(cache_in__, update_in__, valid_seq_len_in__, batch_index_in__, seq_len_axis_in__, new_max_seq_len_in__, cur_max_seq_len_in__, out_out_, kernel_name="prompt_kv_cache", impl_mode=""):
+    # do ascendc build step
     if get_current_build_config("enable_op_prebuild"):
         return
     __inputs__, __outputs__, __attrs__ = _build_args(cache_in__, update_in__, valid_seq_len_in__, batch_index_in__, seq_len_axis_in__, new_max_seq_len_in__, cur_max_seq_len_in__, out_out_)
     options = get_dtype_fmt_options(__inputs__, __outputs__)
     options += ["-x", "cce"]
-    bisheng = shutil.which("bisheng")
+    bisheng = os.environ.get('BISHENG_REAL_PATH')
+    if bisheng is None:
+        bisheng = shutil.which("bisheng")
     if bisheng != None:
         bisheng_path = os.path.dirname(bisheng)
         tikcpp_path = os.path.realpath(os.path.join(bisheng_path, "..", "..", "tikcpp"))
     else:
         tikcpp_path = os.path.realpath("/usr/local/Ascend/latest/compiler/tikcpp")
     options.append("-I" + tikcpp_path)
+    options.append("-I" + os.path.join(tikcpp_path, "..", "..", "include"))
     options.append("-I" + os.path.join(tikcpp_path, "tikcfw"))
     options.append("-I" + os.path.join(tikcpp_path, "tikcfw", "impl"))
     options.append("-I" + os.path.join(tikcpp_path, "tikcfw", "interface"))
@@ -144,8 +174,10 @@ def prompt_kv_cache(cache_in__, update_in__, valid_seq_len_in__, batch_index_in_
         options.append("-DHIGH_PERFORMANCE=1")
     elif impl_mode == "high_precision":
         options.append("-DHIGH_PRECISION=1")
-    if get_default_build_config("enable_deterministic_mode") == 1:
-        options.append("-DDETEMINISTIC_MODE=1")
+    if get_current_build_config("enable_deterministic_mode") == 1:
+        options.append("-DDETERMINISTIC_MODE=1")
+    else:
+        options.append("-DDETERMINISTIC_MODE=0")
 
     custom_compile_options = {},
     custom_all_compile_options = {},
@@ -162,11 +194,14 @@ def prompt_kv_cache(cache_in__, update_in__, valid_seq_len_in__, batch_index_in_
     ascendc_src_file = "prompt_kv_cache.cpp"
     src = get_kernel_source(ascendc_src_file, ascendc_src_dir, ascendc_src_dir_ex)
 
-    print("start compile Ascend C operator PromptKvCache. kernel name is " + kernel_name)
+    msg = "start compile Acend C Operator PromptKvCache, kernel name is " + kernel_name
+    CommonUtility.print_compile_log("", msg, AscendCLogLevel.LOG_INFO)
     op_type = "PromptKvCache"
     code_channel = get_code_channel(src, kernel_name, op_type, options)
     op_info = OpInfo(kernel_name = kernel_name, op_type = op_type, inputs = __inputs__, outputs = __outputs__,\
-        attrs = __attrs__, impl_mode = impl_mode, origin_inputs=[cache_in__, update_in__, valid_seq_len_in__, batch_index_in__, seq_len_axis_in__, new_max_seq_len_in__, cur_max_seq_len_in__], origin_outputs = [out_out_])
+        attrs = __attrs__, impl_mode = impl_mode, origin_inputs=[cache_in__, update_in__, valid_seq_len_in__, batch_index_in__, seq_len_axis_in__, new_max_seq_len_in__, cur_max_seq_len_in__], origin_outputs = [out_out_],\
+                param_type_dynamic = False, mc2_ctx = [], param_type_list = ['required', 'required', 'required', 'required', 'required', 'required', 'required', 'required'], init_value_list = [None],\
+                output_shape_depend_on_compute = [])
     compile_op(src, origin_func_name, op_info, options, code_channel, '{}')
 
 def op_select_format(cache_in__, update_in__, valid_seq_len_in__, batch_index_in__, seq_len_axis_in__, new_max_seq_len_in__, cur_max_seq_len_in__, out_out_, impl_mode=""):