DeepSpeed
Cache Allocation and Softmax Fixes
#2433
Merged

Cache Allocation and Softmax Fixes #2433

jeffra merged 7 commits into master from cholmes/scratch-fix
cmikeh2
cmikeh2 Fix cache, softmax threads tweak
92be033a
fix reduce_width for softmax
de43550f
cmikeh2 cmikeh2 requested a review from jeffra jeffra 3 years ago
cmikeh2 cmikeh2 requested a review from samyam samyam 3 years ago
cmikeh2 cmikeh2 requested a review from tjruwase tjruwase 3 years ago
cmikeh2 cmikeh2 requested a review from ShadenSmith ShadenSmith 3 years ago
cmikeh2 cmikeh2 requested a review from conglongli conglongli 3 years ago
cmikeh2 cmikeh2 requested a review from awan-10 awan-10 3 years ago
cmikeh2 cmikeh2 requested a review from cli99 cli99 3 years ago
cmikeh2 cmikeh2 requested a review from eltonzheng eltonzheng 3 years ago
cmikeh2 cmikeh2 requested a review from minjiaz minjiaz 3 years ago
cmikeh2 cmikeh2 requested a review from RezaYazdaniAminabadi RezaYazdaniAminabadi 3 years ago
cmikeh2 cmikeh2 requested a review from duli2012 duli2012 3 years ago
cmikeh2 cmikeh2 requested a review from mrwyattii mrwyattii 3 years ago
cmikeh2 cmikeh2 requested a review from yaozhewei yaozhewei 3 years ago
cmikeh2 cmikeh2 requested a review from arashb arashb 3 years ago
cmikeh2 cmikeh2 requested a review from xiaoxiawu-microsoft xiaoxiawu-microsoft 3 years ago
cmikeh2 cmikeh2 requested a review from samadejacobs samadejacobs 3 years ago
cmikeh2 cmikeh2 requested a review from GuanhuaWang GuanhuaWang 3 years ago
cmikeh2 cmikeh2 enabled auto-merge (squash) 3 years ago
jeffra Merge branch 'master' into cholmes/scratch-fix
9f295bf0
cmikeh2 Merge branch 'master' into cholmes/scratch-fix
edfb311c
use the user-defined max_tokens for memory-allocation
105e1f82
fix formatting
169762bb
cmikeh2 Merge branch 'master' into cholmes/scratch-fix
3b9081e0
RezaYazdaniAminabadi
RezaYazdaniAminabadi approved these changes on 2022-11-02
RezaYazdaniAminabadi
jeffra
jeffra approved these changes on 2022-11-02
disabled auto-merge 3 years ago
Manually disabled by user
jeffra jeffra merged 10e9d04c into master 3 years ago
jeffra jeffra deleted the cholmes/scratch-fix branch 3 years ago

Login to write a write a comment.

Login via GitHub