3 people 0fbc551622 train bert with fp8 (#13874) ...

* fp8 train

* clean

* lint

* test fix from #13439

* skip first/last layer

* rm __init__, restore unroll <=32 check

* tests

* clean test, remove unused

* multi-gpu test, clean quantize_to_fp8

* remove bert contiguous

* run script

* test: better check

* run script search

* add seed in bert data shuffle

* move script to mi350x folder

---------

Co-authored-by: chenyu <[email protected]>

2026-01-09 09:21:59 -05:00

..

fp8_linear.py

train bert with fp8 (#13874)

2026-01-09 09:21:59 -05:00