onnxruntime
85434273 - Fix CUDA Reduction kernel for ArgMax/ArgMix for when reduction dim=1 (#6490)

Commit

4 years ago

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

References

#6490 - Fix CUDA Reduction kernel for ArgMax/ArgMix for when reduction dim=1

Author

ashbhandare

Parents

14f7d56c

onnxruntime 85434273 - Fix CUDA Reduction kernel for ArgMax/ArgMix for when reduction dim=1 (#6490)

onnxruntime
85434273 - Fix CUDA Reduction kernel for ArgMax/ArgMix for when reduction dim=1 (#6490)