onnxruntime/tools/ci_build/github/pai
ashbhandare 85434273ff
Fix CUDA Reduction kernel for ArgMax/ArgMix for when reduction dim=1 (#6490)
* Fix for when reduction dim=1

* Disable test for AMD GPUs

* Specify Async
2021-02-02 09:50:16 -08:00
..
orttraining-ci.yml
pai-excluded-tests.txt Fix CUDA Reduction kernel for ArgMax/ArgMix for when reduction dim=1 (#6490) 2021-02-02 09:50:16 -08:00
pai_test_launcher.sh
requirements.txt