Skip to content

Commit 8c777f9

Browse files
committed
fix: disable topkrouter on Iluvatar GPU via ENABLE_NVIDIA_API macro
1 parent e8e25a2 commit 8c777f9

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/infiniop/ops/topkrouter/nvidia/topkrouter_nvidia.cu

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
#ifdef ENABLE_NVIDIA_API
2+
13
#include "../../../devices/nvidia/nvidia_common.cuh"
24
#include "../../../devices/nvidia/nvidia_kernel_common.cuh"
35
#include "../cuda/kernel.cuh"
@@ -86,3 +88,5 @@ infiniStatus_t Descriptor::calculate(
8688
return INFINI_STATUS_SUCCESS;
8789
}
8890
} // namespace op::topkrouter::nvidia
91+
92+
#endif

0 commit comments

Comments
 (0)