PR #8511 Cherry pick autocast softmax PR

Cherry pick autocast softmax PR #8511

tengyifei merged 1 commit into r2.6 from cp_autocast_softmax

Added softmax to autocast policy (#8509)

e2d16b6c

avizon-aws requested a review from

cowanmeg 1 year ago

tengyifei added tpuci

tengyifei approved these changes on 2025-01-02

tengyifei merged 7d95b391 into r2.6 1 year ago

Reviewers

tengyifei

cowanmeg

Assignees

No one assigned

Labels

None yet

Milestone

No milestone