onnxruntime/orttraining
edgchen1 dba22b17b4
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.
2020-07-02 18:55:44 -07:00
..
orttraining Update BiasGeluGradDxKernel and tests. (#4400) 2020-07-02 18:55:44 -07:00
pytorch_frontend_examples Remove ORTModel from frontend API (#3825) 2020-05-11 18:20:33 -07:00
tools adjust bs size (#4375) 2020-06-30 10:29:48 -07:00