diff --git a/src/ops_infer/self_attention/self_attention_operation.cpp b/src/ops_infer/self_attention/self_attention_operation.cpp index e7262743b6f50f168029486924a6fc56af87d5f1..56174f9ab33c061b68b6cb435c694c65a8d0254f 100644 --- a/src/ops_infer/self_attention/self_attention_operation.cpp +++ b/src/ops_infer/self_attention/self_attention_operation.cpp @@ -57,7 +57,7 @@ template <> Status CreateOperation(const infer::SelfAttentionParam &opParam, Ope << ", clampType:" << opParam.clampType << ", kvcacheCfg:" << opParam.kvcacheCfg << ", scaleType:" << opParam.scaleType << ", inputLayout:" << opParam.inputLayout << ", mlaVHeadSize:" << opParam.mlaVHeadSize << ", windowSize:" << opParam.windowSize - << ", cacheType: " << opParam.cacheType; + << ", cacheType: " << opParam.cacheType << ", kernelType:" << opParam.kernelType; if (opParam.headNum <= 0) { ATB_LOG(ERROR) << "headNum should be greater than zero!"; return ERROR_INVALID_PARAM;