From 02315a8dbe74afeda8951ae727d8553e53c7eb55 Mon Sep 17 00:00:00 2001
From: Charles Xu <charles.xu@arm.com>
Date: Mon, 10 Feb 2025 15:32:06 +0100
Subject: [PATCH] Move ggml-cpu-traits.h to source file

---
 ggml/src/ggml-cpu/kleidiai/kernels.h    | 2 ++
 ggml/src/ggml-cpu/kleidiai/kleidiai.cpp | 1 +
 ggml/src/ggml-cpu/kleidiai/kleidiai.h   | 2 +-
 3 files changed, 4 insertions(+), 1 deletion(-)

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" {