onnxruntime
34725ae5
- Bugfix for topk cuda kernel (#6164)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
5 years ago
Bugfix for topk cuda kernel (#6164) * fix the issue that std::numeric_limits cannot handle half type * adding a test Co-authored-by: Du Li <duli@OrtTrainingDev4.af05slrtruoetgaxwwjv5nsq5e.px.internal.cloudapp.net>
References
#6164 - Bugfix for topk cuda kernel
Author
duli2012
Parents
dec703b6
Loading