DeepSpeed
681be6f5 - Fix CPU Adam JIT compilation (#5780)

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>
Author
Parents
Loading