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

Merged

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

ashbhandare merged 3 commits into master from aibhanda/reduce_single

Fix for when reduction dim=1

091771ab

ashbhandare requested a review 5 years ago

ashbhandare added core runtime

ashbhandare added training

ashbhandare removed training

ashbhandare added ep:CUDA

weixingzhang dismissed these changes on 2021-01-28

Disable test for AMD GPUs

3bcac4ac

ashbhandare dismissed their stale review via 3bcac4ac 5 years ago

jessebenson dismissed these changes on 2021-01-29

Specify Async

eff7b47b

ashbhandare dismissed their stale review via eff7b47b 5 years ago

weixingzhang approved these changes on 2021-02-02

ashbhandare merged 85434273 into master 5 years ago

ashbhandare deleted the aibhanda/reduce_single branch 5 years ago

Reviewers

weixingzhang

jessebenson

Assignees

No one assigned

Labels

core runtime ep:CUDA

Milestone

No milestone