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

MLAS: fix ARM64 + VS2017 build break (#5423)

Browse source
This commit is contained in:
Tracy Sharpe 2020-10-08 18:03:45 -07:00 committed by GitHub
parent 8a632a903f
commit fab7f799a7
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
onnxruntime/core/mlas/lib/qgemm.cpp
View file

@ -1721,7 +1721,17 @@ Return Value:
RowSums = vpadalq_u16(RowSums, vpaddlq_u8(v));
}
#if defined(_M_ARM64)
// N.B. The workaround of defining a local vaddvq_u32 doesn't work here
// as VS2019 added new intrinsics to make the operation work. Also, not
// all build environments using VS2019 have the up-to-date arm64_neon.h,
// so fallback to pairwise addition.
RowSums = vpaddq_u32(RowSums, RowSums);
RowSums = vpaddq_u32(RowSums, RowSums);
vst1q_lane_u32(reinterpret_cast<int32_t*>(RowSumBuffer), RowSums, 0);
#else
*RowSumBuffer = int32_t(vaddvq_u32(RowSums));
#endif
}
}