[ROCm] Make DSA indexer top-k exact with cooperative selection (#37591)
This commit is contained in:
+1
-1
@@ -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",
|
||||
|
||||
Reference in New Issue
Block a user