Public
  1. Public

onnxruntime

Public

Commits

ashbhandareGitHub
14f7d56c81b
ashbhandare authored and GitHub committed 85434273ffe
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