onnxruntime
72c689a5
- [CUDA] Use dim3.z to Handle Large Input For GatherGrad (#12250)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
3 years ago
[CUDA] Use dim3.z to Handle Large Input For GatherGrad (#12250) * use dim3.z to handle large input size * less blocks
References
#12250 - [CUDA] Use dim3.z to Handle Large Input For GatherGrad
Author
Lafi7e
Parents
ebfd81e6
Loading