onnxruntime
Rashuai/cuda opt top k
#2093
Merged
Overview
Commits
10
Changes
View On GitHub

Rashuai/cuda opt top k #2093

RandySheriffH merged 10 commits into CudaOpt from rashuai/CudaOptTopK
RandySheriffH
RandyShuai implement cuda topk
e01ad99b
RandyShuai implement heap
c2c4bc47
RandyShuai add type support
b95426aa
RandyShuai refactor interface
084cd0f7
RandyShuai add support for sorting by index
92e39bf8
RandyShuai add test case
93eae964
RandyShuai merge with master
681fca6b
RandyShuai use cub device radix sort
351972fa
RandyShuai register for opset 9 and 10
675a7e56
RandyShuai merge topk cuda
e6cb72f4
RandySheriffH RandySheriffH requested a review 7 years ago
RandySheriffH RandySheriffH merged d02d95b8 into CudaOpt 7 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum