Skip to content

Commit b876860

Browse files
authored
[Hardware][CPU] Build fix for ARM without BF16 (#21848)
Signed-off-by: Eric Curtin <[email protected]>
1 parent 1398636 commit b876860

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

csrc/cpu/quant.cpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,12 +16,14 @@ struct KernelVecType<float> {
1616
using cvt_vec_type = vec_op::FP32Vec16;
1717
};
1818

19+
#if !defined(__aarch64__) || defined(ARM_BF16_SUPPORT)
1920
template <>
2021
struct KernelVecType<c10::BFloat16> {
2122
using load_vec_type = vec_op::BF16Vec16;
2223
using azp_adj_load_vec_type = vec_op::INT32Vec16;
2324
using cvt_vec_type = vec_op::FP32Vec16;
2425
};
26+
#endif
2527

2628
template <>
2729
struct KernelVecType<c10::Half> {

0 commit comments

Comments
 (0)