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
|
d7bb62cfa1
|
refactor(mgb): move mm_handler from python module into opr-mm
GitOrigin-RevId: f401ce8603
|
5 years ago |
Megvii Engine Team
|
dbf7764012
|
fix(mge/mm_handler): fix unused var ins
GitOrigin-RevId: 199d084e8f
|
5 years ago |
Megvii Engine Team
|
0293d58ade
|
feat(mge): add bfloat16 support
GitOrigin-RevId: a942ce6791
|
6 years ago |
Megvii Engine Team
|
c3d5b61f1c
|
feat(mgb/opr): add property params for python operator class
GitOrigin-RevId: af6da0e0ac
|
5 years ago |
Megvii Engine Team
|
396653df5f
|
feat(mge/api): expose sublinear related parameters at mge api level
GitOrigin-RevId: 7a47f5d0d5
|
5 years ago |
Megvii Engine Team
|
b8cbd4510b
|
fix(mgb/mm_server): return -1 when create_mm_server failed
GitOrigin-RevId: 117bb80c06
|
5 years ago |
Megvii Engine Team
|
cec7fd3c5f
|
fix(mgb/build): fix multi-machine macro and add test_distributed
GitOrigin-RevId: cb1bfe8742
|
5 years ago |
Megvii Engine Team
|
3fce8544ec
|
refactor(mgb/parampacksplit): remove input offsets on device
GitOrigin-RevId: b2bf3bf155
|
5 years ago |
Megvii Engine Team
|
5121626e98
|
refactor(dnn/parampack): remove parampacksplit
GitOrigin-RevId: ed00341d58
|
5 years ago |
Megvii Engine Team
|
b708f15d39
|
refactor(mgb/param_pack): use shared mem for param pack
GitOrigin-RevId: bc56f09037
|
5 years ago |
Megvii Engine Team
|
f18259d73a
|
refactor(mge/zmqrpc): zmqrpc return real port or -1
GitOrigin-RevId: 878b63c601
|
5 years ago |
Megvii Engine Team
|
dbe2b89331
|
fix(midout): fix brain opr midout 2/2 (also see a6aa1574)
fix extern_c_opr midout
GitOrigin-RevId: 7de4f650d1
|
5 years ago |
Megvii Engine Team
|
fc6aa12e2f
|
feat(mge):add qint4/quint4 to python
GitOrigin-RevId: f94609db00
|
5 years ago |
Megvii Engine Team
|
646829b802
|
ci(pytest): use common method install pip package
GitOrigin-RevId: 87162b8b56
|
5 years ago |
Megvii Engine Team
|
9edce7e294
|
ci(python_test): run python test with cuda
GitOrigin-RevId: bc51c2e4d5
|
5 years ago |
Megvii Engine Team
|
f91881ffdc
|
MegEngine: Initial commit of MegEngine.
GitOrigin-RevId: f0c8338beb
|
6 years ago |