Commit 8c777f97 authored by zhushuang's avatar zhushuang
Browse files

fix: disable topkrouter on Iluvatar GPU via ENABLE_NVIDIA_API macro

parent e8e25a25
#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