onnxruntime
dba22b17 - Update BiasGeluGradDxKernel and tests. (#4400)

Commit
5 years ago
Update BiasGeluGradDxKernel and tests. (#4400) For BiasGeluGradDxKernel: - Implement optimization to first load from global memory into registers as suggested by Weixing. - Support larger bias sizes which were previously limited by the number of threads per block. - Address flaky unit test by increasing the error tolerance to the default value.
Author
Parents
Loading