From 74ae0d1acedbd4de1640062004bd6fcc9f35e636 Mon Sep 17 00:00:00 2001
From: Ruihang Lai <ruihangl@cs.cmu.edu>
Date: Tue, 18 Mar 2025 10:40:38 -0400
Subject: [PATCH] [Fix] Fix OpenCL header in attention utils

This PR fixes the OpenCL header include for `attn_utils.h`. The include
was not moved from `paged_kv_cache.cc` when separating out the file
`attn_utils.h`.
---
 src/runtime/relax_vm/attn_utils.h      | 3 +++
 src/runtime/relax_vm/paged_kv_cache.cc | 3 ---
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/runtime/relax_vm/attn_utils.h b/src/runtime/relax_vm/attn_utils.h
index af46073adefe..8138aa7bbdf6 100644
--- a/src/runtime/relax_vm/attn_utils.h
+++ b/src/runtime/relax_vm/attn_utils.h
@@ -30,6 +30,9 @@
 #include <limits>
 #include <utility>
 #include <vector>
+#if defined(OPENCL_ENABLE_HOST_PTR)
+#include "../opencl/opencl_common.h"
+#endif
 
 namespace tvm {
 namespace runtime {
diff --git a/src/runtime/relax_vm/paged_kv_cache.cc b/src/runtime/relax_vm/paged_kv_cache.cc
index 6b68bb1b1a30..496891f2172b 100644
--- a/src/runtime/relax_vm/paged_kv_cache.cc
+++ b/src/runtime/relax_vm/paged_kv_cache.cc
@@ -36,9 +36,6 @@
 #include "attn_backend.h"
 #include "attn_utils.h"
 #include "kv_state.h"
-#if defined(OPENCL_ENABLE_HOST_PTR)
-#include "../opencl/opencl_common.h"
-#endif
 
 namespace tvm {
 namespace runtime {