We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e8e25a2 commit 8c777f9Copy full SHA for 8c777f9
src/infiniop/ops/topkrouter/nvidia/topkrouter_nvidia.cu
@@ -1,3 +1,5 @@
1
+#ifdef ENABLE_NVIDIA_API
2
+
3
#include "../../../devices/nvidia/nvidia_common.cuh"
4
#include "../../../devices/nvidia/nvidia_kernel_common.cuh"
5
#include "../cuda/kernel.cuh"
@@ -86,3 +88,5 @@ infiniStatus_t Descriptor::calculate(
86
88
return INFINI_STATUS_SUCCESS;
87
89
}
90
} // namespace op::topkrouter::nvidia
91
92
+#endif
0 commit comments