From df6a0f5d47f86b7713780383a3824e70b847120f Mon Sep 17 00:00:00 2001
From: claude <claude@anthropic.com>
Date: Fri, 14 Aug 2026 11:29:19 +0000
Subject: [PATCH] fix: remove cuda/functional from MoE topk kernels (not
 available on corex)

---
 ex_engine/xllm_kernels/cuda/moe/moe_topk_sigmoid_kernels.cuh | 1 -
 ex_engine/xllm_kernels/cuda/moe/moe_topk_softmax_kernels.cuh | 1 -
 2 files changed, 2 deletions(-)

diff --git a/ex_engine/xllm_kernels/cuda/moe/moe_topk_sigmoid_kernels.cuh b/ex_engine/xllm_kernels/cuda/moe/moe_topk_sigmoid_kernels.cuh
index 68e22daf..15b4dc90 100644
--- a/ex_engine/xllm_kernels/cuda/moe/moe_topk_sigmoid_kernels.cuh
+++ b/ex_engine/xllm_kernels/cuda/moe/moe_topk_sigmoid_kernels.cuh
@@ -24,7 +24,6 @@ limitations under the License.
 #include <cub/util_type.cuh>
 
 #if !defined(USE_DCU) && !defined(USE_MACA)
-#include <cuda/functional>
 #endif
 
 #include "kernels/cuda/device_utils.cuh"
diff --git a/ex_engine/xllm_kernels/cuda/moe/moe_topk_softmax_kernels.cuh b/ex_engine/xllm_kernels/cuda/moe/moe_topk_softmax_kernels.cuh
index 4dea9aba..43119a37 100644
--- a/ex_engine/xllm_kernels/cuda/moe/moe_topk_softmax_kernels.cuh
+++ b/ex_engine/xllm_kernels/cuda/moe/moe_topk_softmax_kernels.cuh
@@ -24,7 +24,6 @@ limitations under the License.
 #include <cub/util_type.cuh>
 
 #if !defined(USE_DCU) && !defined(USE_MACA)
-#include <cuda/functional>
 #endif
 
 #include "kernels/cuda/device_utils.cuh"