From 8c777f975cdadb376771425ce6bdfee216d81154 Mon Sep 17 00:00:00 2001 From: zhushuang <974198603@qq.com> Date: Tue, 16 Sep 2025 15:02:03 +0800 Subject: [PATCH] fix: disable topkrouter on Iluvatar GPU via ENABLE_NVIDIA_API macro --- src/infiniop/ops/topkrouter/nvidia/topkrouter_nvidia.cu | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/infiniop/ops/topkrouter/nvidia/topkrouter_nvidia.cu b/src/infiniop/ops/topkrouter/nvidia/topkrouter_nvidia.cu index 5ba9ddc62..e44872fcc 100644 --- a/src/infiniop/ops/topkrouter/nvidia/topkrouter_nvidia.cu +++ b/src/infiniop/ops/topkrouter/nvidia/topkrouter_nvidia.cu @@ -1,3 +1,5 @@ +#ifdef ENABLE_NVIDIA_API + #include "../../../devices/nvidia/nvidia_common.cuh" #include "../../../devices/nvidia/nvidia_kernel_common.cuh" #include "../cuda/kernel.cuh" @@ -86,3 +88,5 @@ infiniStatus_t Descriptor::calculate( return INFINI_STATUS_SUCCESS; } } // namespace op::topkrouter::nvidia + +#endif