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

Exclude fp16 support flag definition from minimal build (#17259)

Browse source 
### Description
<!-- Describe your changes. -->

As title.

### Motivation and Context
<!-- - Why is this change required? What problem does it solve?
- If it fixes an open issue, please link to the issue here. -->

Reduce minimal build binary size for mobile to meet office team
requirement.

cc @chenfucn

Co-authored-by: rachguo <rachguo@rachguos-Mini.attlocal.net>
This commit is contained in:
Rachel Guo 2023-08-23 10:13:19 -07:00 committed by GitHub
parent b3cb775cf9
commit aed7c6ffc7
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/core/mlas/inc/mlas.h
View file

@ -77,6 +77,7 @@ Abstract:
#define MLAS_SUPPORTS_GEMM_DOUBLE
#endif
#if !defined(ORT_MINIMAL_BUILD) || defined(ORT_EXTENDED_MINIMAL_BUILD)
#if (!defined(_MSC_VER)) || (_MSC_VER >= 1930)
#if defined(MLAS_TARGET_ARM64) || defined(MLAS_TARGET_ARM64EC)
#if !defined(__APPLE__)
@ -90,6 +91,7 @@ Abstract:
#endif //
#endif // ARM64
#endif // Visual Studio 16 or earlier does not support fp16 intrinsic
#endif // !defined(ORT_MINIMAL_BUILD) || defined(ORT_EXTENDED_MINIMAL_BUILD)
//
// Basic Linear Algebra Subprograms (BLAS) types.

4
onnxruntime/core/mlas/lib/halfgemm_kernel_neon.cpp
View file

@ -14,6 +14,8 @@ Abstract:
--*/
#if !defined(ORT_MINIMAL_BUILD) || defined(ORT_EXTENDED_MINIMAL_BUILD)
#include "mlasi.h"
#include "halfgemm.h"
@ -185,3 +187,5 @@ const MLAS_HALFGEMM_DISPATCH MlasHalfGemmDispatchNeon = {
MLAS_HALF_GEMM_KERNEL_NEON::KernelMaxM,
32 // kernel may read beyond buffer end by 32 bytes
};
#endif // !defined(ORT_MINIMAL_BUILD) || defined(ORT_EXTENDED_MINIMAL_BUILD)