PR #7133 [Pallas] Make gmm support bf16

[Pallas] Make gmm support bf16 #7133

alanwaketan merged 3 commits into master from alanwaketan/gmm7

Torchify the reference gmm

24c176ad

fix bf16

0b7bc394

Fix linters

b187387a

alanwaketan added tpuci

alanwaketan requested a review from

miladm 1 year ago

alanwaketan requested a review from

wonjoo-wj 1 year ago

alanwaketan requested a review from

JackCaoG 1 year ago

alanwaketan assigned

alanwaketan 1 year ago

JackCaoG commented on 2024-05-29

JackCaoG approved these changes on 2024-05-29

alanwaketan merged fb373129 into master 1 year ago

alanwaketan deleted the alanwaketan/gmm7 branch 1 year ago

Reviewers

JackCaoG

miladm

wonjoo-wj

Assignees

alanwaketan

Labels

None yet

Milestone

No milestone