optimum-habana
optimum-habana copied to clipboard
Initial enablement with FP8 Training (port from OHF #91)
What does this PR do?
Initial enablement with FP8 Training with Intel Gaudi Transformer Engine ([(porting from OHF #91) Only linear layer is replaced with FP8.
Fixes # (issue)
Before submitting
- [ ] This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case).
- [ ] Did you make sure to update the documentation with your changes?
- [ ] Did you write any new necessary tests?