Skip to content

Commit 12a859f

Browse files
committed
fix compile
1 parent ae9cb82 commit 12a859f

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

csrc/cpu/torch_bindings.cpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -91,6 +91,7 @@ TORCH_LIBRARY_EXPAND(TORCH_EXTENSION_NAME, ops) {
9191
ops.impl("rotary_embedding", torch::kCPU, &rotary_embedding);
9292

9393
// Quantization
94+
#ifdef __AVX512F__
9495
// Compute int8 quantized tensor for given scaling factor.
9596
ops.def(
9697
"static_scaled_int8_quant(Tensor! out, Tensor input, Tensor scale) -> "
@@ -109,6 +110,7 @@ TORCH_LIBRARY_EXPAND(TORCH_EXTENSION_NAME, ops) {
109110
" Tensor b, Tensor a_scales,"
110111
" Tensor b_scales, Tensor? bias) -> ()");
111112
ops.impl("cutlass_scaled_mm", torch::kCPU, &int8_scaled_mm);
113+
#endif
112114
}
113115

114116
TORCH_LIBRARY_EXPAND(CONCAT(TORCH_EXTENSION_NAME, _cache_ops), cache_ops) {

0 commit comments

Comments
 (0)