onnxruntime/orttraining
pengwa 6e09fc5152
Implement block wise softmax for reduction dimention > 1024 cases. (#9696)
* implement block wise softmax for reduction dimention > 1024 cases.

* fix builds

* fix

* fix amd build

* fix amd build

* fix win-gpu build

* add tests

* remove cudnn path/add python tests
2021-11-14 11:47:58 +08:00
..
orttraining Implement block wise softmax for reduction dimention > 1024 cases. (#9696) 2021-11-14 11:47:58 +08:00
pytorch_frontend_examples Sync ORTModule branch with master and fix tests (#6526) 2021-02-02 08:59:56 -08:00
tools Switch AMD CI pipeline to use environment image from onnxruntimecibuildenvironment (#9206) 2021-09-28 13:06:16 -07:00