Address review comments

vllm-project · Jun 12, 2024 · 4db9a59 · 4db9a59
1 parent ae4803c
commit 4db9a59
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/csrc/cpu/cpu_types.hpp b/csrc/cpu/cpu_types.hpp
@@ -5,6 +5,10 @@
 #include <immintrin.h>
 #include <torch/all.h>
 
+#ifndef __AVX2__
+static_assert(false, "AVX2 must be supported for the current implementation.");
+#endif
+
 namespace vec_op {
 
 // FIXME: FP16 is not fully supported in Torch-CPU