PR #14541 [ROCm] add Softmax Tunable Op

[ROCm] add Softmax Tunable Op #14541

PeixuanZuo merged 8 commits into main from peixuanzuo/update_rocm_softmax

PeixuanZuo force pushed from 3f329f8a to 62d42877 3 years ago

PeixuanZuo requested a review from

cloudhan 3 years ago

PeixuanZuo requested a review from

abudup 3 years ago

PeixuanZuo requested a review from

zhangyaobit 3 years ago

Base automatically changed from peixuanzuo/fix_softmax_block_forward to main 3 years ago

PeixuanZuo force pushed from 62d42877 to ee0b6324 3 years ago

abudup commented on 2023-02-09

simplify warpwise_softmax_forward code

9daecece

add tunable softmax block

892101c0

add SoftmaxTunableOp

fa627f63

add ck softmax

eb1a641e

update format

817923e9

rebase

3354bbd6

address comments

10a7350e

add TuningContext

0b6c029f

PeixuanZuo force pushed from ee0b6324 to 0b6c029f 3 years ago

abudup approved these changes on 2023-02-10

PeixuanZuo merged 326cf2f5 into main 3 years ago

PeixuanZuo deleted the peixuanzuo/update_rocm_softmax branch 3 years ago

Reviewers

abudup

cloudhan

zhangyaobit

Assignees

No one assigned

Labels

None yet

Milestone

No milestone