saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-17 21:10:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/tools/ci_build/github/pai
History
Suffian Khan 76bc0e479c
Enable dense sequence optimized version of Pytorch exported BERT-L on AMD GPU (#6504) 
* Permit dense seq optimization on BERT-L pytorch export by enabling ReduceSumTraining, Equal, and NonZero on AMD

* enable Equal tests

* enable fast_matrix_reduction test case
2021-01-29 13:12:34 -08:00
..
orttraining-ci.yml ROCm EP for AMD GPU (#5480) 2020-10-29 17:13:04 -07:00
pai-excluded-tests.txt Enable dense sequence optimized version of Pytorch exported BERT-L on AMD GPU (#6504) 2021-01-29 13:12:34 -08:00
pai_test_launcher.sh ROCm EP for AMD GPU (#5480) 2020-10-29 17:13:04 -07:00
requirements.txt ROCm EP for AMD GPU (#5480) 2020-10-29 17:13:04 -07:00