diff --git a/ggml/src/ggml-cpu/kleidiai/kernels.h b/ggml/src/ggml-cpu/kleidiai/kernels.h
index f606eb2ef..2ffe97eb4 100644
--- a/ggml/src/ggml-cpu/kleidiai/kernels.h
+++ b/ggml/src/ggml-cpu/kleidiai/kernels.h
@@ -2,6 +2,8 @@
 // SPDX-License-Identifier: MIT
 //
 
+#pragma once
+
 enum cpu_feature {
     CPU_FEATURE_NONE    = 0,
     CPU_FEATURE_DOTPROD = 1,
diff --git a/ggml/src/ggml-cpu/kleidiai/kleidiai.cpp b/ggml/src/ggml-cpu/kleidiai/kleidiai.cpp
index c9235cd63..2c7413ff5 100644
--- a/ggml/src/ggml-cpu/kleidiai/kleidiai.cpp
+++ b/ggml/src/ggml-cpu/kleidiai/kleidiai.cpp
@@ -24,6 +24,7 @@
 #include "ggml-impl.h"
 #include "ggml-backend-impl.h"
 #include "ggml-threading.h"
+#include "ggml-cpu-traits.h"
 
 #include "kernels.h"
 
diff --git a/ggml/src/ggml-cpu/kleidiai/kleidiai.h b/ggml/src/ggml-cpu/kleidiai/kleidiai.h
index 6fd6b257a..38eac58f7 100644
--- a/ggml/src/ggml-cpu/kleidiai/kleidiai.h
+++ b/ggml/src/ggml-cpu/kleidiai/kleidiai.h
@@ -4,7 +4,7 @@
 
 #pragma once
 
-#include "ggml-cpu-traits.h"
+#include "ggml-alloc.h"
 
 #ifdef  __cplusplus
 extern "C" {