Skip to content

Commit

Permalink
Sqnbitgemm: add loongarch64 code path (microsoft#18775)
Browse files Browse the repository at this point in the history
### Description

Add support code for loongarch64 platform in sqnbitgemm

```
100% tests passed, 0 tests failed out of 7

Total Test time (real) = 116.99 sec
2023-12-11 10:43:21,287 build [INFO] - Build complete

```
  • Loading branch information
junchao-loongson authored Jan 9, 2024
1 parent dee6a5b commit c1367ae
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion onnxruntime/core/mlas/lib/sqnbitgemm.h
Original file line number Diff line number Diff line change
Expand Up @@ -232,7 +232,7 @@ MlasSQNBitGemmOperation(

size_t RowsRemaining = RangeCountM;
while (RowsRemaining > 0) {
#if defined(MLAS_TARGET_AMD64_IX86) || defined(MLAS_TARGET_POWER)
#if defined(MLAS_TARGET_AMD64_IX86) || defined(MLAS_TARGET_POWER) || defined(MLAS_TARGET_LARCH64)
auto RowsHandled = GetMlasPlatform().GemmFloatKernel(
a_row, dequant_b, c_blk, K, RowsRemaining, CountN, lda, ldc, 1.f, true
);
Expand Down

0 comments on commit c1367ae

Please sign in to comment.