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
Hide 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> {
using cvt_vec_type = vec_op::FP32Vec16;
};
#if !defined(__aarch64__) || defined(ARM_BF16_SUPPORT)
template <>
struct KernelVecType<c10::BFloat16> {
using load_vec_type = vec_op::BF16Vec16;
using azp_adj_load_vec_type = vec_op::INT32Vec16;
using cvt_vec_type = vec_op::FP32Vec16;
};
#endif
template <>
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