(tvm) branch main updated: [Fix] Fix OpenCL header in attention utils (#17762)

[hongyij]

This is an automated email from the ASF dual-hosted git repository.

hongyij pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new b0f1433bae [Fix] Fix OpenCL header in attention utils (#17762)
b0f1433bae is described below

commit b0f1433bae8d73ba61bda21664dbc4b192f8f31d
Author: Ruihang Lai <ruiha...@cs.cmu.edu>
AuthorDate: Tue Mar 18 15:09:46 2025 -0400

    [Fix] Fix OpenCL header in attention utils (#17762)
    
    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 af46073ade..8138aa7bbd 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 6b68bb1b1a..496891f217 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 {