accelerate
0e4419b3
- Add bf16/fp16 support for amp with mps device (#3373)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
133 days ago
Add bf16/fp16 support for amp with mps device (#3373) * Fix tests * format * amp mps support for fp16/bf16 * add error * revert * revert * fix * ruff
References
#3373 - Add bf16/fp16 support for amp with mps device
Author
SunMarc
Parents
3b67c216
Loading