onnxruntime
Update CUDA TopK kernel registration to opset 24 with BFloat16 support
#27735

Merged

Update CUDA TopK kernel registration to opset 24 with BFloat16 support #27735

tianleiwu merged 6 commits into main from copilot/update-topk-fill-opset-gap

Update CUDA TopK to fill opset gap: cap opset 11 to 11-23, add opset 24

c911bf4a

copilot-swe-agent assigned

copilot-swe-agent 240 days ago

copilot-swe-agent assigned

tianleiwu 240 days ago

Merge remote-tracking branch 'origin/main' into copilot/update-topk-f…

ef1079e2

Update docs/OperatorKernels.md for CUDA TopK opset gap (11+ → [11,23]…

291d8af7

copilot-swe-agent requested a review from

tianleiwu 240 days ago

tianleiwu marked this pull request as ready for review 240 days ago

tianleiwu approved these changes on 2026-03-20

Add BFloat16 support for CUDA TopK opset 24

58775e48

copilot-swe-agent changed the title ~~Update CUDA TopK kernel registration to opset 24~~ Update CUDA TopK kernel registration to opset 24 with BFloat16 support 238 days ago

copilot-swe-agent requested a review from

tianleiwu 238 days ago

Add BFloat16 test cases for TopK opset 24

3ea02338

Fix CUB radix sort build error: map BFloat16 to __nv_bfloat16 for CUB…

e33539c1

tianleiwu approved these changes on 2026-03-24

tianleiwu requested a review from

kunal-vaishnavi 235 days ago

tianleiwu requested a review from

titaiwangms 235 days ago

kunal-vaishnavi approved these changes on 2026-03-24

titaiwangms approved these changes on 2026-03-24

tianleiwu merged a8424e67 into main 234 days ago

tianleiwu deleted the copilot/update-topk-fill-opset-gap branch 234 days ago

Reviewers

titaiwangms

kunal-vaishnavi

tianleiwu

Assignees

tianleiwu

Copilot

Labels

None yet

Milestone

No milestone

onnxruntime Update CUDA TopK kernel registration to opset 24 with BFloat16 support #27735 Merged

Update CUDA TopK kernel registration to opset 24 with BFloat16 support #27735

onnxruntime
Update CUDA TopK kernel registration to opset 24 with BFloat16 support
#27735

Merged