PR #12469 Softmax interface update

Softmax interface update #12469

ytaous merged 5 commits into microsoft:main from ROCm:softmax_interface_update

Template datatype for SoftmaxWithRawMaskSmallKernel in ROCm EP

2610c4dd

Remove valid_items usage from SoftmaxWithRawMaskSmallKernel for ROCm EP

8796d425

Update accumulator type in ROCm EP for SoftmaxWithRawMaskSmallKernel

513cceed

groenenboomj marked this pull request as ready for review 3 years ago

justinchuby requested a review from

edgchen1 3 years ago

justinchuby removed review request from

edgchen1 3 years ago

Reset casting to old behavior

94a886de

tianleiwu dismissed these changes on 2022-08-31

Document steps to optimize SoftMax kernel on ROCm EP

e6380a5a

groenenboomj dismissed their stale review via e6380a5a 3 years ago

zhangyaobit approved these changes on 2022-09-08

ytaous merged a433f22f into main 3 years ago

Reviewers

zhangyaobit

tianleiwu

Assignees

No one assigned

Labels

None yet

Milestone

No milestone