saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-19 19:00:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

use wasmscalar as default kernel (#9988)

Browse source 
* use wasmscalar as default kernel
This commit is contained in:
Yufeng Li 2022-01-03 10:55:08 -08:00 committed by GitHub
parent 28ce2a5a78
commit 7208fcbe1c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 12 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
cmake/onnxruntime_mlas.cmake
View file

@ -176,7 +176,7 @@ if (onnxruntime_BUILD_WEBASSEMBLY)
)
else()
file(GLOB_RECURSE mlas_platform_srcs
"${MLAS_SRC_DIR}/wasm/*.cpp"
"${MLAS_SRC_DIR}/scalar/*.cpp"
)
endif()
target_sources(onnxruntime_mlas PRIVATE ${mlas_platform_srcs})
@ -336,7 +336,7 @@ else()
${mlas_platform_srcs_power10}
)
endif()
endif()
endif()
if(NOT ONNXRUNTIME_MLAS_MULTI_ARCH)
set(MLAS_SOURCE_IS_NOT_SET 0)
endif()
@ -367,7 +367,7 @@ else()
${mlas_platform_srcs}
${MLAS_SRC_DIR}/x86/x86.get_pc_thunk.S
)
endif()
endif()
if(NOT ONNXRUNTIME_MLAS_MULTI_ARCH)
set(MLAS_SOURCE_IS_NOT_SET 0)
@ -462,9 +462,12 @@ else()
list(APPEND ONNXRUNTIME_MLAS_LIBS onnxruntime_mlas_x86_64)
set(mlas_platform_srcs )
else()
set(MLAS_SOURCE_IS_NOT_SET 1)
set(MLAS_SOURCE_IS_NOT_SET 0)
endif()
endif()
if(NOT ONNXRUNTIME_MLAS_MULTI_ARCH AND MLAS_SOURCE_IS_NOT_SET)
file(GLOB_RECURSE mlas_platform_srcs
"${MLAS_SRC_DIR}/scalar/*.cpp")
endif()
target_sources(onnxruntime_mlas PRIVATE ${mlas_platform_srcs})
endif()

2
onnxruntime/core/mlas/lib/wasm/SconvDepthwiseKernelWasmScalar.cpp → onnxruntime/core/mlas/lib/scalar/SconvDepthwiseKernelScalar.cpp
View file

@ -6,7 +6,7 @@ Licensed under the MIT License.
Module Name:
SconvDepthwiseKernelWasmScalar.cpp
SconvDepthwiseKernelScalar.cpp
Abstract:

4
onnxruntime/core/mlas/lib/wasm/SgemmKernelWasmScalar.cpp → onnxruntime/core/mlas/lib/scalar/SgemmKernelScalar.cpp
View file

@ -6,7 +6,7 @@ Licensed under the MIT License.
Module Name:
SgemmKernelWasmScalar.cpp
SgemmKernelScalar.cpp
Abstract:
@ -41,7 +41,7 @@ Arguments:
A - Supplies the address of matrix A.
B - Supplies the address of matrix B. The matrix data has been packed using
MlasSgemmCopyPackB or MlasSgemmTransposePackB. Note that in WASM scalar,
MlasSgemmCopyPackB or MlasSgemmTransposePackB. Note that in scalar,
the packing wide is 4.
C - Supplies the address of matrix C.

2
onnxruntime/core/mlas/lib/wasm/SgemvKernelWasmScalar.cpp → onnxruntime/core/mlas/lib/scalar/SgemvKernelScalar.cpp
View file

@ -6,7 +6,7 @@ Licensed under the MIT License.
Module Name:
SgemvKernelWasmScalar.cpp
SgemvKernelScalar.cpp
Abstract: