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 ROCm EP for AMD GPU (#5480) 2020-10-29 17:13:04 -07:00
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 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