PR #21045 Zhijxu/fix softmax cudnn bf16

Zhijxu/fix softmax cudnn bf16 #21045

zhijxu-MS merged 6 commits into main from zhijxu/fix-softmax-cudnn-bf16

github-advanced-security commented on 2024-06-14

zhijxu-MS force pushed from a13f132c to 395ee828 1 year ago

make Uniform support BFloat16 dtype

003a85db

add large softmaxgrad bf16 test

c01eb7ce

fix GetDataType<BFloat16>

65326186

fix lint

94cf43e0

skipt bf16 test when gpu not supports bf16 such as v100

8999a9cb

zhijxu-MS force pushed from 395ee828 to 8999a9cb 1 year ago

wschin commented on 2024-06-21

wschin dismissed these changes on 2024-06-21

fix review comments

cb6a07a9

zhijxu-MS dismissed their stale review via cb6a07a9 1 year ago

Lafi7e approved these changes on 2024-06-24

zhijxu-MS merged 269d9b09 into main 1 year ago

zhijxu-MS deleted the zhijxu/fix-softmax-cudnn-bf16 branch 1 year ago

Reviewers

Lafi7e

wschin

Assignees

No one assigned

Labels

None yet

Milestone

No milestone