[AMD] Enable deepseek-v4 topk_transform v2 kernel - #36684

Merged

HaiShaw merged 10 commits into

sgl-project:mainfrom

1am9trash:v4-topk-v2

Aug 28, 2026

Commits on Aug 27, 2026

Remove cooperative_groups.h import
1am9trash
committed
f8e518e
Fix kFullMask in __shfl_up_sync()
1am9trash
committed
4570cc9
Fix __ballot() 64-bit mismatch to __popc()
1am9trash
committed
32ff1b5
Comment TopKCluster part on ROCm
1am9trash
committed
4cb4e38
Comment Cluster part on ROCm in topk_v2.cuh
1am9trash
committed
803c2b0
Change kDLCUDA to kDLGPU
1am9trash
committed
68c436e
Comment use_cluster branch
1am9trash
committed
71eaf4d
Enable v2 gate on amd code path
1am9trash
committed
99f86f4
Enable amd CI test
1am9trash
committed
013000e
Correct gate setting
1am9trash
committed
527723a