PR #10398 [ROCm] Enable BFloat16 for Gemm and MatMul Op

[ROCm] Enable BFloat16 for Gemm and MatMul Op #10398

ytaous merged 5 commits into master from ettao/bf16

gemm-bf16

ae3a3c98

gemm bf16

ecd308ca

gemm bf16

76b80283

ytaous changed the title ~~WIP - [ROCm] Enable BFloat16 for Gemm~~ [ROCm] Enable BFloat16 for Gemm Op 3 years ago

ytaous requested a review from

weixingzhang 3 years ago

ytaous requested a review from

Lafi7e 3 years ago

ytaous added training

matmul bf16

cc49ad52

ytaous changed the title ~~[ROCm] Enable BFloat16 for Gemm Op~~ [ROCm] Enable BFloat16 for Gemm and MatMul Op 3 years ago

minor style change

45edc791

Lafi7e approved these changes on 2022-01-27

weixingzhang approved these changes on 2022-01-27

ytaous merged 4d305282 into master 3 years ago

ytaous deleted the ettao/bf16 branch 3 years ago

Reviewers

Lafi7e

weixingzhang

Assignees

No one assigned

Labels

training

Milestone

No milestone