diff --git a/src/ops_infer/multi_latent_attention/multi_latent_attention_operation.cpp b/src/ops_infer/multi_latent_attention/multi_latent_attention_operation.cpp index 02949a59047a49d436b1c3ae8cfe4162d5dc14e2..3eaf2b3ef68363287efcce715a07f3a2d9b694e1 100644 --- a/src/ops_infer/multi_latent_attention/multi_latent_attention_operation.cpp +++ b/src/ops_infer/multi_latent_attention/multi_latent_attention_operation.cpp @@ -109,8 +109,9 @@ static bool ParamCheck(const infer::MultiLatentAttentionParam &opParam) if ((opParam.calcType == infer::MultiLatentAttentionParam::CalcType::CALC_TYPE_RING || opParam.calcType == infer::MultiLatentAttentionParam::CalcType::CALC_TYPE_SPEC_AND_RING) && (opParam.cacheMode != infer::MultiLatentAttentionParam::CacheMode::KROPE_CTKV && - opParam.cacheMode != infer::MultiLatentAttentionParam::CacheMode::INT8_NZCACHE)) { - ATB_LOG(ERROR) << "CalcType is ring only support krppe ctkv and int8 nzcache"; + opParam.cacheMode != infer::MultiLatentAttentionParam::CacheMode::INT8_NZCACHE && + opParam.cacheMode != infer::MultiLatentAttentionParam::CacheMode::NZCACHE)) { + ATB_LOG(ERROR) << "CalcType is ring only support krope, ctkv, int8 nzcache and nzcache"; return false; } if ((opParam.cacheMode == infer::MultiLatentAttentionParam::CacheMode::INT8_NZCACHE) &&