Megvii Engine Team
|
27205461ae
|
feat(mgb/opr-mm): add register info cache for multi-machine oprs
GitOrigin-RevId: d5ae3c5a7c
|
4 years ago |
Megvii Engine Team
|
80c4705317
|
perf(mgb): use midout in megbrain to reduce binary size
GitOrigin-RevId: ddc8af79af
|
4 years ago |
Megvii Engine Team
|
5e912eddbd
|
fix(mgb/opr-mm): fix grad func of reduce and gather
GitOrigin-RevId: 4687faef99
|
4 years ago |
Megvii Engine Team
|
e3e981ccf0
|
test(mge/collective_comm): fix collective_comm test and add data parallel test
GitOrigin-RevId: 9209e77973
|
4 years ago |
Megvii Engine Team
|
d3d9018f8d
|
feat(imperative/opr-mm): add broadcast
GitOrigin-RevId: 83640255c7
|
4 years ago |
Megvii Engine Team
|
6d367454cf
|
feat(mge/opr-mm): add param local_grad for collective_comm opr
GitOrigin-RevId: cc120cfb55
|
5 years ago |
Megvii Engine Team
|
328fb36f17
|
feat(mgb/opr-mm): add Scatter, Gather, AllToAll oprs
GitOrigin-RevId: f75169ecd6
|
5 years ago |
Megvii Engine Team
|
01092feb9b
|
feat(mgb): add PackAllReducePass
GitOrigin-RevId: 59c1b45393
|
5 years ago |
Megvii Engine Team
|
c7e6c658fd
|
refactor(mge/distribute): use is_root (and rank) in stead of rank and root at collective comm
GitOrigin-RevId: dccdb71553
|
5 years ago |
Megvii Engine Team
|
cde055e8f4
|
fix(mgb/opr-mm): fix user_data thread safety in CollectiveComm
GitOrigin-RevId: b6d6184e91
|
5 years ago |
Megvii Engine Team
|
2bbce2f924
|
fix(mgb/opr-mm): fix m_megray_ctx init
GitOrigin-RevId: 7804fbe2ef
|
5 years ago |
Megvii Engine Team
|
b87a9a74b1
|
feat(mgb/opr-mm): add debug log for CollectiveComm opr
GitOrigin-RevId: 11bcf11043
|
5 years ago |
Megvii Engine Team
|
f91881ffdc
|
MegEngine: Initial commit of MegEngine.
GitOrigin-RevId: f0c8338beb
|
5 years ago |