Skip to content

GitLab

Unverified Commit b876860c authored by Eric Curtin's avatar Eric Curtin Committed by GitHub
Browse files

[Hardware][CPU] Build fix for ARM without BF16 (#21848) 


Signed-off-by: default avatarEric Curtin <ecurtin@redhat.com>
parent 13986365
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
csrc/cpu/quant.cpp
View file @ b876860c
...@@ -16,12 +16,14 @@ struct KernelVecType<float> { ...@@ -16,12 +16,14 @@ struct KernelVecType<float> {
using cvt_vec_type = vec_op::FP32Vec16; using cvt_vec_type = vec_op::FP32Vec16;
}; };
#if !defined(__aarch64__) || defined(ARM_BF16_SUPPORT)
template <> template <>
struct KernelVecType<c10::BFloat16> { struct KernelVecType<c10::BFloat16> {
using load_vec_type = vec_op::BF16Vec16; using load_vec_type = vec_op::BF16Vec16;
using azp_adj_load_vec_type = vec_op::INT32Vec16; using azp_adj_load_vec_type = vec_op::INT32Vec16;
using cvt_vec_type = vec_op::FP32Vec16; using cvt_vec_type = vec_op::FP32Vec16;
}; };
#endif
template <> template <>
struct KernelVecType<c10::Half> { struct KernelVecType<c10::Half> {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment