DeepSpeed
Fix potential random layout inconsistency issues in sparse attention modules
#534
Merged

Fix potential random layout inconsistency issues in sparse attention modules #534

arashashari merged 7 commits into deepspeedai:master from Justin1904:master
Justin1904
1) Register layout as buffer of module so that we can save/load check…
fb28742b
Justin1904 Justin1904 requested a review from arashashari arashashari 5 years ago
Justin1904 Justin1904 requested a review from awan-10 awan-10 5 years ago
Justin1904 Justin1904 requested a review from cli99 cli99 5 years ago
Justin1904 Justin1904 requested a review from conglongli conglongli 5 years ago
Justin1904 Justin1904 requested a review from eltonzheng eltonzheng 5 years ago
Justin1904 Justin1904 requested a review from jeffra jeffra 5 years ago
Justin1904 Justin1904 requested a review from minjiaz minjiaz 5 years ago
Justin1904 Justin1904 requested a review from niumanar niumanar 5 years ago
Justin1904 Justin1904 requested a review from RezaYazdaniAminabadi RezaYazdaniAminabadi 5 years ago
Justin1904 Justin1904 requested a review from samyam samyam 5 years ago
Justin1904 Justin1904 requested a review from ShadenSmith ShadenSmith 5 years ago
Justin1904 Justin1904 requested a review from tjruwase tjruwase 5 years ago
arashashari
arashashari commented on 2020-11-18
arashashari
arashashari approved these changes on 2020-11-18
Add docstring for max_seq_length argument in SparseSelfAttention
2509bad0
jeffra
jeffra commented on 2020-11-18
jeffra Merge branch 'master' into master
2a3e0a8f
jeffra Merge branch 'master' into master
e087d99f
jeffra Merge branch 'master' into master
da2dd1e0
jeffra Merge branch 'master' into master
fbdf2bfc
jeffra Merge branch 'master' into master
8bed4178
arashashari arashashari merged 1e44d48d into master 5 years ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone