[ROCm] enable moe topk softmax in amd (#4448)

2025-03-16 18:16:55 -07:00
parent f5bbf6037d
commit 9b8333d992
2 changed files with 5 additions and 0 deletions
@@ -41,6 +41,7 @@ include_dirs = [
 sources = [
    "csrc/allreduce/custom_all_reduce.hip",
    "csrc/moe/moe_align_kernel.cu",
+    "csrc/moe/moe_topk_softmax_kernels.cu",
    "csrc/torch_extension_rocm.cc",
 ]