已合并
feat(dequant_swiglu_quant): swiglu_mode参数校验扩展支持3 #5811
caorenlei创建于 28 天前
feat(dequant_swiglu_quant): swiglu_mode参数校验扩展支持3 #5811
已合并
共 1 个文件变更+2-2
| @@ -47,8 +47,8 @@ std::tuple<at::Tensor, at::Tensor> npu_dequant_swiglu_quant( | |||
| 47 | quant_mode, | 47 | quant_mode, |
| 48 | OPS_ERROR(ErrCode::PARAM)); | 48 | OPS_ERROR(ErrCode::PARAM)); |
| 49 | TORCH_CHECK( | 49 | TORCH_CHECK( |
| 50 | - swiglu_mode == 0 || swiglu_mode == 1 || swiglu_mode == 2, | 50 | + swiglu_mode == 0 || swiglu_mode == 1 || swiglu_mode == 2 || swiglu_mode == 3, |
| 51 | - "swiglu_mode only support 0, 1 or 2, but got ", | 51 | + "swiglu_mode only support 0, 1 , 2 or 3, but got ", |
| 52 | swiglu_mode, | 52 | swiglu_mode, |
| 53 | OPS_ERROR(ErrCode::PARAM)); | 53 | OPS_ERROR(ErrCode::PARAM)); |
| 54 | TORCH_CHECK( | 54 | TORCH_CHECK( |