Megvii Engine Team
|
369c2ccc5a
|
style(all): reformat c++ code
GitOrigin-RevId: 3ffd1b211f
|
4 years ago |
Megvii Engine Team
|
0b4a767965
|
feat(mge/distributed): enable uint8 for collective communication
GitOrigin-RevId: 3305c0cf14
|
5 years ago |
Megvii Engine Team
|
809d5056cd
|
feat(mge/distributed): enable pt shm allreduce
GitOrigin-RevId: 1dd5a02a51
|
5 years ago |
Megvii Engine Team
|
1a7112997c
|
feat(opr-mm): add backend argument for remote send/recv
GitOrigin-RevId: 841a0e45ab
|
5 years ago |
Megvii Engine Team
|
1bec737d1a
|
feat(distributed): support distributed opr for rocm
GitOrigin-RevId: 4840100d07
|
5 years ago |
Megvii Engine Team
|
6af0299c3f
|
test(mge/ioremote): add REQUIRE_GPU for ioremote opr test
GitOrigin-RevId: 31ec8982c4
|
5 years ago |
Megvii Engine Team
|
364afec033
|
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
|
5 years ago |
Megvii Engine Team
|
2ad8c5e1e9
|
fix(mge/io_remote): fix remote send/recv gradient at trace
GitOrigin-RevId: 7886efd0c1
|
5 years ago |
Megvii Engine Team
|
064c774f05
|
feat(imperative): impl hashable for SendRecv and add virtual input for Recv
GitOrigin-RevId: 5e8c27ac81
|
5 years ago |
Megvii Engine Team
|
ca717806af
|
fix(mgb): fix wrong use of macro MGB_ENABLE_GRAD
GitOrigin-RevId: e66dabee01
|
5 years ago |
Megvii Engine Team
|
4a5e317063
|
fix(mgb/opr-mm): remove stream -4 for CollectiveComm
GitOrigin-RevId: 41ea88dfa1
|
5 years ago |
Megvii Engine Team
|
116eee5231
|
build(third_party): update megray
GitOrigin-RevId: da5e05f82b
|
5 years ago |
Megvii Engine Team
|
50d5421aa7
|
fix(mgb/zmq): fix unused-result when compiling with c++17
GitOrigin-RevId: cbdffd39c6
|
5 years ago |
Megvii Engine Team
|
1bce857cb8
|
fix(mgb/opr-mm): use comp_node of config as default in CollectiveComm
GitOrigin-RevId: 6b43c9fc93
|
6 years ago |
Megvii Engine Team
|
27205461ae
|
feat(mgb/opr-mm): add register info cache for multi-machine oprs
GitOrigin-RevId: d5ae3c5a7c
|
6 years ago |
Megvii Engine Team
|
f7b5eced23
|
refactor(mgb/opr-mm): set False as default value of local_grad
GitOrigin-RevId: 2f9603b087
|
6 years ago |
Megvii Engine Team
|
80c4705317
|
perf(mgb): use midout in megbrain to reduce binary size
GitOrigin-RevId: ddc8af79af
|
6 years ago |
Megvii Engine Team
|
be5dde9da8
|
fix(mgb/opr-mm): fix group manager sync error
GitOrigin-RevId: 344aae534c
|
6 years ago |
Megvii Engine Team
|
4cf193ab97
|
fix(mgb/opr-mm): fix device id in TestOprIORemote.IdentityWithGopt
GitOrigin-RevId: 0184036b8b
|
6 years ago |
Megvii Engine Team
|
c76789a5c9
|
feat(mge/imperative): add io remote wrapper
GitOrigin-RevId: 6dae8287d3
|
6 years ago |
Megvii Engine Team
|
b38e8225be
|
refactor(mgb/opr-mm): update megray communicator init interface and fix ci
GitOrigin-RevId: 55c59879f2
|
6 years ago |
Megvii Engine Team
|
5e912eddbd
|
fix(mgb/opr-mm): fix grad func of reduce and gather
GitOrigin-RevId: 4687faef99
|
6 years ago |
Megvii Engine Team
|
e3e981ccf0
|
test(mge/collective_comm): fix collective_comm test and add data parallel test
GitOrigin-RevId: 9209e77973
|
6 years ago |
Megvii Engine Team
|
d3d9018f8d
|
feat(imperative/opr-mm): add broadcast
GitOrigin-RevId: 83640255c7
|
6 years ago |
Megvii Engine Team
|
cf3a55ce17
|
fix(mgb/opr-mm): remove PeerDesc from RemoteSend and RemoteRecv
GitOrigin-RevId: b7a7bbd0da
|
6 years ago |
Megvii Engine Team
|
6d367454cf
|
feat(mge/opr-mm): add param local_grad for collective_comm opr
GitOrigin-RevId: cc120cfb55
|
6 years ago |
Megvii Engine Team
|
328fb36f17
|
feat(mgb/opr-mm): add Scatter, Gather, AllToAll oprs
GitOrigin-RevId: f75169ecd6
|
6 years ago |
Megvii Engine Team
|
01092feb9b
|
feat(mgb): add PackAllReducePass
GitOrigin-RevId: 59c1b45393
|
6 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
|
6 years ago |
Megvii Engine Team
|
d7bb62cfa1
|
refactor(mgb): move mm_handler from python module into opr-mm
GitOrigin-RevId: f401ce8603
|
6 years ago |
Megvii Engine Team
|
cde055e8f4
|
fix(mgb/opr-mm): fix user_data thread safety in CollectiveComm
GitOrigin-RevId: b6d6184e91
|
6 years ago |
Megvii Engine Team
|
26242d6bab
|
test(mgb/opr-mm): add collective_comm tests
GitOrigin-RevId: 02edb26ba7
|
6 years ago |
Megvii Engine Team
|
cd8ab9e3a6
|
test(mgb/opr-mm): add io_remote test
GitOrigin-RevId: c47b6156fe
|
6 years ago |
Megvii Engine Team
|
4e0054f7b2
|
fix(mgb/opr-mm): fix megray_helper thread safety
GitOrigin-RevId: f7b7c1d97f
|
6 years ago |
Megvii Engine Team
|
2bbce2f924
|
fix(mgb/opr-mm): fix m_megray_ctx init
GitOrigin-RevId: 7804fbe2ef
|
6 years ago |
Megvii Engine Team
|
b87a9a74b1
|
feat(mgb/opr-mm): add debug log for CollectiveComm opr
GitOrigin-RevId: 11bcf11043
|
6 years ago |
Megvii Engine Team
|
f91881ffdc
|
MegEngine: Initial commit of MegEngine.
GitOrigin-RevId: f0c8338beb
|
6 years ago |