PR #3322 Add fp8 custom op and unit test

Add fp8 custom op and unit test #3322

copybara-service merged 5 commits into google:main from wenscarl:shuw/add_fp8_support

kaixih commented on 2023-09-11

levskaya commented on 2023-09-21

wenscarl requested a review from

levskaya 2 years ago

wenscarl requested a review from

kaixih 2 years ago

levskaya approved these changes on 2023-09-22

Add fp8 custom op and unit test

b309c8f2

Clean imports and add train_states

2bea7a57

Add TrainState

04d71b65

self.params replaces variable_with_axes

06f5f280

Fix Fp8 train state test

b1ee11fc

wenscarl force pushed to b1ee11fc 2 years ago

wenscarl requested a review from

levskaya 2 years ago

levskaya added pull ready

copybara-service merged cf7d89a3 into main 2 years ago

Reviewers

levskaya

kaixih

Assignees

No one assigned

Labels

pull ready

Milestone

No milestone