onnxruntime
Bugfix for topk cuda kernel
#6164
Merged

Bugfix for topk cuda kernel #6164

duli2012 merged 2 commits into master from duli/topk
duli2012
duli2012 fix the issue that std::numeric_limits cannot handle half type
8490e0a3
duli2012 duli2012 requested a review from HectorSVC HectorSVC 5 years ago
duli2012 duli2012 requested a review from hanbitmyths hanbitmyths 5 years ago
duli2012 duli2012 requested a review 5 years ago
duli2012 duli2012 requested a review from ke1337 ke1337 5 years ago
hanbitmyths
adding a test
61947074
duli2012
hanbitmyths
hanbitmyths approved these changes on 2020-12-17
duli2012 duli2012 merged 34725ae5 into master 5 years ago
duli2012 duli2012 deleted the duli/topk branch 5 years ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone