Skip to content

Commit 4e916c5

Browse files
author
yulinjie
committed
feature: some bug option fix.
Change-Id: Ic3dddae8c8f89559a13513cd77b25d01b285a40e
1 parent 241c357 commit 4e916c5

File tree

2 files changed

+3
-5
lines changed

2 files changed

+3
-5
lines changed

ggml/src/ggml-cpu/onnxruntime_mlas/CMakeLists.txt

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,6 @@ elseif(NOT CMAKE_SYSTEM_NAME STREQUAL "Emscripten")
5959
set(ARM64 TRUE)
6060
elseif(CMAKE_SYSTEM_PROCESSOR MATCHES "^riscv64.*")
6161
set(RISCV64 TRUE)
62-
set(RISCV64_SPACEMIT_IME_SPEC RISCV64_SPACEMIT_IME2)
6362
elseif(CMAKE_SYSTEM_PROCESSOR MATCHES "^(i.86|x86?)$")
6463
set(X86 TRUE)
6564
elseif(CMAKE_SYSTEM_PROCESSOR MATCHES "^(x86_64|amd64)$")
@@ -777,9 +776,7 @@ endif()
777776
${MLAS_SRC_DIR}/sqnbitgemm_kernel_spacemit_ime_int8.cpp
778777
${MLAS_SRC_DIR}/sqnbitgemm_kernel_spacemit_ime_fp32.cpp
779778
)
780-
endif()
781-
782-
if (RISCV64_SPACEMIT_IME_SPEC STREQUAL "RISCV64_SPACEMIT_IME2")
779+
elseif (RISCV64_SPACEMIT_IME_SPEC STREQUAL "RISCV64_SPACEMIT_IME2")
783780
target_compile_definitions(onnxruntime_mlas PRIVATE ${RISCV64_SPACEMIT_IME_SPEC})
784781
set(mlas_platform_srcs
785782
${mlas_platform_srcs}

scripts/build-riscv64-spacemit.sh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,11 +7,12 @@ SCRIPTS_DIR=$(cd "$(dirname "$(readlink -f "$0")")" && pwd)
77
cmake -B build-riscv64-spacemit \
88
-DCMAKE_BUILD_TYPE=Release \
99
-DGGML_CPU_RISCV64_SPACEMIT=ON \
10+
-DRISCV64_SPACEMIT_IME_SPEC=RISCV64_SPACEMIT_IME1 \
1011
-DCMAKE_TOOLCHAIN_FILE=${SCRIPTS_DIR}/../cmake/riscv64-spacemit-linux-gnu-gcc.cmake \
1112
-DCMAKE_INSTALL_PREFIX=build-riscv64-spacemit/installed
1213

1314
cmake --build build-riscv64-spacemit --parallel 10 --config Release
1415

1516
pushd build-riscv64-spacemit
1617
make install
17-
popd
18+
popd

0 commit comments

Comments
 (0)