[Piecewise] Support PCG weak_ref_tensor cuda kernel on AMD (#17291)

2026-01-20 14:05:32 -08:00
parent 20ed3822bb
commit 38c233fd04
3 changed files with 9 additions and 2 deletions
@@ -53,6 +53,7 @@ sources = [
    "csrc/moe/moe_topk_sigmoid_kernels.cu",
    "csrc/speculative/eagle_utils.cu",
    "csrc/kvcacheio/transfer.cu",
+    "csrc/memory/weak_ref_tensor.cpp",
    "csrc/elementwise/pos_enc.cu",
    "csrc/sgl_diffusion/elementwise/timestep_embedding.cu",
 ]