Skip to content

Commit 25d58f0

Browse files
authored
fix: add missing BackendKernelSelectorConfig parameter to QNBitGemmPackQuantBDataSize_Lasx (#28179)
### Description PR #27136 introduced a new `const MLAS_BACKEND_KERNEL_SELECTOR_CONFIG*` parameter to the `Q4BitGemmPackQuantBDataSize_Fn` typedef in `MLAS_QNBIT_GEMM_DISPATCH`, and updated the three `PackQuantBData*` functions in `sqnbitgemm_kernel_lasx.cpp` accordingly. However, `QNBitGemmPackQuantBDataSize_Lasx` was missed, leaving it with a 5-parameter signature that no longer matches the typedef. This causes a compile error on LoongArch LASX builds when the function's address is assigned to `d.Q4BitGemmPackQuantBDataSize`. This change adds the missing 6th parameter to `QNBitGemmPackQuantBDataSize_Lasx`, matching the signature used by the three sibling functions in the same file (parameter name commented out since it is unused by this implementation). ### Motivation and Context Fixes #28157. Restores the LoongArch LASX build without altering runtime behavior on any platform. ### Changes - `onnxruntime/core/mlas/lib/sqnbitgemm_kernel_lasx.cpp`: add `const MLAS_BACKEND_KERNEL_SELECTOR_CONFIG* /*BackendKernelSelectorConfig*/` as the 6th parameter of `QNBitGemmPackQuantBDataSize_Lasx`. ### Test Plan - The change is a signature-only fix with no behavioral delta; the added parameter is unused (consistent with the three sibling functions in the same file). - Existing x86 and ARM CI should continue to pass since no logic was changed. - LoongArch LASX build, which previously failed, now matches the dispatch typedef and compiles.
1 parent 5743f71 commit 25d58f0

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

onnxruntime/core/mlas/lib/sqnbitgemm_kernel_lasx.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,8 @@ QNBitGemmPackQuantBDataSize_Lasx(
3232
size_t K,
3333
size_t BlkLen,
3434
bool /* HasZeroPoint */,
35-
MLAS_QNBIT_GEMM_COMPUTE_TYPE ComputeType
35+
MLAS_QNBIT_GEMM_COMPUTE_TYPE ComputeType,
36+
const MLAS_BACKEND_KERNEL_SELECTOR_CONFIG* /*BackendKernelSelectorConfig*/
3637
)
3738
{
3839
const size_t BlockCountK = MlasDivRoundup(K, BlkLen);

0 commit comments

Comments
 (0)