DeepSpeed
681be6f5
- Fix CPU Adam JIT compilation (#5780)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
Fix CPU Adam JIT compilation (#5780) This PR fixes CPU Adam JIT compilation by including the `CUDA_LIB64` path in the `extra_ldflags` list before calling `load()`. --------- Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com>
References
#5780 - Fix CPU Adam JIT compilation
Author
lekurile
Parents
550f9c75
Loading