[ROCm] Make DSA indexer top-k exact with cooperative selection (#37591)

This commit is contained in:
Zhang, Jiejing
2026-09-05 23:55:56 -07:00
committed by GitHub
parent e3f7097591
commit 6cee9285a3
5 changed files with 1672 additions and 2 deletions
+1 -1
View File
@@ -18,7 +18,7 @@ sources = [
"csrc/common_extension_rocm.cc",
"csrc/elementwise/activation.cu",
"csrc/elementwise/pos_enc.cu",
"csrc/elementwise/topk.cu",
# topk.hip is maintained as native HIP instead of being generated from topk.cu.
"csrc/grammar/apply_token_bitmask_inplace_cuda.cu",
"csrc/kvcacheio/transfer.cu",
"csrc/memory/weak_ref_tensor.cpp",