PR #10085 CUDA BFloat16 Refactor

CUDA BFloat16 Refactor #10085

Lafi7e merged 10 commits into master from weicwang/bfloat16

Use BFLoat16 instead of nv_bfloat16 for CUDA

2b3f8f7d

fix BFloat16 rocm

8815c589

remove gemm ut from rocm

5467e8e4

remove reducesum ut from rocm

2cfec118

add biasgelu

2b8868ee

add support for more ops

7dc37da4

Lafi7e added training

Lafi7e requested a review from

SherlockNoMad 4 years ago

Lafi7e requested a review from

weixingzhang 4 years ago

Merge branch 'master' into weicwang/bfloat16

d071935a

fix build error

a3c394af

op md update

882cb356

weixingzhang commented on 2022-01-06

Merge branch 'master' into weicwang/bfloat16

8c3240dc

weixingzhang approved these changes on 2022-01-14

Lafi7e merged 44e2db93 into master 4 years ago

Lafi7e deleted the weicwang/bfloat16 branch 4 years ago

Reviewers

weixingzhang

SherlockNoMad

Assignees

No one assigned

Labels

training

Milestone

No milestone