diff --git a/src/ailego/math/norm1_matrix_fp16.cc b/src/ailego/math/norm1_matrix_fp16.cc
index 9bb8620..e75b3e0 100644
--- a/src/ailego/math/norm1_matrix_fp16.cc
+++ b/src/ailego/math/norm1_matrix_fp16.cc
@@ -13,6 +13,7 @@
 // limitations under the License.
 
 #include <zvec/ailego/internal/platform.h>
+#include "ailego/internal/cpu_features.h"
 #include "norm1_matrix.h"
 #include "norm_matrix_fp16.i"
 
diff --git a/src/ailego/math/norm1_matrix_fp32.cc b/src/ailego/math/norm1_matrix_fp32.cc
index c1ec366..2e72791 100644
--- a/src/ailego/math/norm1_matrix_fp32.cc
+++ b/src/ailego/math/norm1_matrix_fp32.cc
@@ -13,6 +13,7 @@
 // limitations under the License.
 
 #include <zvec/ailego/internal/platform.h>
+#include "ailego/internal/cpu_features.h"
 #include "norm1_matrix.h"
 #include "norm_matrix_fp32.i"
 
diff --git a/src/ailego/math/norm2_matrix_fp16.cc b/src/ailego/math/norm2_matrix_fp16.cc
index 37c3313..6bb8dd0 100644
--- a/src/ailego/math/norm2_matrix_fp16.cc
+++ b/src/ailego/math/norm2_matrix_fp16.cc
@@ -13,6 +13,7 @@
 // limitations under the License.
 
 #include <zvec/ailego/internal/platform.h>
+#include "ailego/internal/cpu_features.h"
 #include "norm2_matrix.h"
 #include "norm_matrix_fp16.i"