Megvii Engine Team
|
09241a1ff7
|
feat(mge): remove param_pack_* from functional
GitOrigin-RevId: a5fe25be8c
|
4 years ago |
Megvii Engine Team
|
6c5cf25f4d
|
docs(mge/distributed): add distributed.helper docs
GitOrigin-RevId: 37c14aa11f
|
4 years ago |
Megvii Engine Team
|
36a4fb5611
|
ci(image): update docker image install python package
GitOrigin-RevId: 171e95b3d9
|
4 years ago |
Megvii Engine Team
|
1a24fb29c1
|
perf(mge/allreduce): put allreduce on another cuda stream
GitOrigin-RevId: 2e778dfa04
|
4 years ago |
Megvii Engine Team
|
aac8de554e
|
fix(mge/parampacksplit): fix parampacksplit refcnt error
GitOrigin-RevId: c964465596
|
4 years ago |
Megvii Engine Team
|
8d02d10483
|
refactor(mge/distributed): change bcast_params_ to bcast_list_
GitOrigin-RevId: 26b452a6b7
|
4 years ago |
Megvii Engine Team
|
c7acba41fc
|
refactor(mge/optimizer): refine gradmanager api, record = __enter__
GitOrigin-RevId: 5376177237
|
4 years ago |
Megvii Engine Team
|
8c482b6709
|
fix(mge/grad): make register_after_backward_callback private
GitOrigin-RevId: 8eb6c0e628
|
4 years ago |
Megvii Engine Team
|
e9104ef157
|
fix(mge/parampack): fix copy stream, import cycle
GitOrigin-RevId: 673e11c5b6
|
4 years ago |
Megvii Engine Team
|
b5016b9d29
|
feat(mge/parampack): add parampack in allreduce callback
GitOrigin-RevId: 73d53eeba1
|
4 years ago |
Megvii Engine Team
|
6070266766
|
refactor(mge/grad_manager): refactor gradmanager, add allreduce callback
GitOrigin-RevId: 086e2871e8
|
4 years ago |
Megvii Engine Team
|
e1fba6ece7
|
test(mge/distributed): add get_device_count_by_fork to fix distributed test skip
GitOrigin-RevId: 9ffd8a6149
|
4 years ago |
Megvii Engine Team
|
6b380e8965
|
feat(mge/imperative): run oss test and restore cmake list build items
GitOrigin-RevId: 11411b6964
|
4 years ago |