Skip to content

GitLab

Unverified Commit 1f507406 authored by PanZezhong1725's avatar PanZezhong1725 Committed by GitHub
Browse files

fix: disable topkrouter on Iluvatar GPU via ENABLE_NVIDIA_API macro (#457)

parents b9dd0004 8c777f97
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
src/infiniop/ops/topkrouter/nvidia/topkrouter_nvidia.cu
View file @ 1f507406
#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
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment