We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent fbe324c commit c980b28Copy full SHA for c980b28
csrc/cpu/quant.cpp
@@ -16,12 +16,14 @@ struct KernelVecType<float> {
16
using cvt_vec_type = vec_op::FP32Vec16;
17
};
18
19
+#if !defined(__aarch64__) || defined(ARM_BF16_SUPPORT)
20
template <>
21
struct KernelVecType<c10::BFloat16> {
22
using load_vec_type = vec_op::BF16Vec16;
23
using azp_adj_load_vec_type = vec_op::INT32Vec16;
24
25
26
+#endif
27
28
29
struct KernelVecType<c10::Half> {
0 commit comments