DeepSpeed
DeepSpeed JIT op + PyPI support
#496
Merged
Overview
Commits
23
Changes
View On GitHub

DeepSpeed JIT op + PyPI support #496

jeffra merged 23 commits into master from jeffra/jit_extensions_v2
jeffra
jeffra jeffra requested a review from arashashari arashashari 5 years ago
jeffra jeffra requested a review from awan-10 awan-10 5 years ago
jeffra jeffra requested a review from cli99 cli99 5 years ago
jeffra jeffra requested a review from conglongli conglongli 5 years ago
jeffra jeffra requested a review from eltonzheng eltonzheng 5 years ago
jeffra jeffra requested a review from minjiaz minjiaz 5 years ago
jeffra jeffra requested a review from niumanar niumanar 5 years ago
jeffra jeffra requested a review from RezaYazdaniAminabadi RezaYazdaniAminabadi 5 years ago
jeffra jeffra requested a review from samyam samyam 5 years ago
jeffra jeffra requested a review from ShadenSmith ShadenSmith 5 years ago
jeffra jeffra requested a review from tjruwase tjruwase 5 years ago
jeffra (squash) JIT extensions (#494)
73daba6e
jeffra formatting
4d44d16c
jeffra updates to install and fix apex import in tests
22be69d5
jeffra remove deepspeed_adam
4a85e99d
jeffra fixes for cpu adam avx
c753bd3d
jeffra update pypi badge and text around ops
d6ff46b9
jeffra fix doc links for install details
79b2160f
jeffra update install reqs if op is compatible
dff70b86
jeffra fixes and tested
3e3a8bab
jeffra jeffra force pushed from ba5a99d0 to 3e3a8bab 5 years ago
jeffra add note to env report
fbb9576c
jeffra formatting
5fda8cd6
jeffra debugging cpu-adam issue, forcing pre-install of ops
4dba7751
jeffra remove cc 52, causing build issues
425504dc
replace memory-format with device and dtype parameters
ae06d188
change parameter-update to do single casting at each thread
3b745c56
jeffra switch tests back to use JIT op
9f6cb2ae
jeffra Merge branch 'jeffra/jit_extensions_v2' of github.com:microsoft/DeepS…
ae48706c
jeffra Merge branch 'master' into jeffra/jit_extensions_v2
24827fc4
jeffra update setup to allow for pip install build ops
bfe8ef0c
jeffra Merge branch 'master' into jeffra/jit_extensions_v2
c2842378
jeffra Update index.md
aeeee2c7
jeffra Update README.md
21172dc8
tjruwase
tjruwase approved these changes on 2020-11-12
ShadenSmith
ShadenSmith approved these changes on 2020-11-12
remove comments and increase threads for parameter-update for CPU-Adam
9956bce1
RezaYazdaniAminabadi
RezaYazdaniAminabadi approved these changes on 2020-11-12
jeffra jeffra merged 31f46fee into master 5 years ago
jeffra jeffra deleted the jeffra/jit_extensions_v2 branch 5 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum