Megvii Engine Team
|
809d5056cd
|
feat(mge/distributed): enable pt shm allreduce
GitOrigin-RevId: 1dd5a02a51
|
4 years ago |
Megvii Engine Team
|
933dd9a497
|
feat(mge/distributed): add cuda env check before forked thread
style(core/comp_node): reformat code
GitOrigin-RevId: 372452a8eb
|
4 years ago |
Megvii Engine Team
|
0cf4ff702e
|
perf(dist): add fastpath for bcast params
GitOrigin-RevId: aa40b3cd72
|
4 years ago |
Megvii Engine Team
|
dcfb6a537e
|
refactor(mge/functional): move functional api
GitOrigin-RevId: 9cd3e09996
|
4 years ago |
Megvii Engine Team
|
fa4bf16800
|
feat(mge/functional): add repeat and tile opr
GitOrigin-RevId: a20d4b6fb0
|
4 years ago |
Megvii Engine Team
|
364afec033
|
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
|
4 years ago |
Megvii Engine Team
|
b9762d714c
|
fix(mge): make parampack run with tensor symbolic shape
GitOrigin-RevId: 6fc313785d
|
4 years ago |
Megvii Engine Team
|
60c7d62a4c
|
refactor(imperative): remove multidispatch, raw_tensor, register
GitOrigin-RevId: ca5a6ed8eb
|
4 years ago |
Megvii Engine Team
|
147cef52a0
|
refactor(mge/imperative): implement new tensor system
GitOrigin-RevId: 2dd4e460ac
|
4 years ago |
Megvii Engine Team
|
638ab52fdc
|
feat(mge/imperative): simulates scalar
GitOrigin-RevId: e81630e256
|
4 years ago |
Megvii Engine Team
|
6667100638
|
feat(mge): use weakref for GradManger.attach
GitOrigin-RevId: 6df336c3c1
|
4 years ago |
Megvii Engine Team
|
a1efd4d56d
|
fix(mge/parampack): fix param pack when no param left
GitOrigin-RevId: b88b876064
|
4 years ago |
Megvii Engine Team
|
b309890c66
|
docs(mge): pytest for sphinx docstring
GitOrigin-RevId: 8bed12562a
|
4 years ago |
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 |