[AMD] Expand test coverage for AMD CI and enable apply_token_bitmask_inplace_cuda in sgl-kernel (#8268)

2025-08-15 12:32:51 -07:00
parent e52c3866eb
commit 9c3e95d98b
6 changed files with 61 additions and 6 deletions
--- a/sgl-kernel/setup_rocm.py
+++ b/sgl-kernel/setup_rocm.py
@@ -48,6 +48,7 @@ sources = [
    "csrc/moe/moe_topk_softmax_kernels.cu",
    "csrc/speculative/eagle_utils.cu",
    "csrc/common_extension_rocm.cc",
+    "csrc/grammar/apply_token_bitmask_inplace_cuda.cu",
 ]

 cxx_flags = ["-O3"]