Megvii Engine Team
05ee603843
fix(mge/distributed): fix single_machine_cache error
GitOrigin-RevId: be57111408
3 years ago
Megvii Engine Team
dff7719e87
feat(mge/distributed): add preload host data with op
fix(mge/distributed): change api name with preload
fix(mge/distributed): fix recursive model in preload tensor
fix(mge/distributed): fix recursive when cache contain None
GitOrigin-RevId: 80e2a6dd70
3 years ago
Megvii Engine Team
c5f8b58385
docs(docstring): transfer to google style
GitOrigin-RevId: a71245c553
3 years ago
Megvii Engine Team
5fe789abb9
fix(mge/distributed): skip check when machine_ranks is None
GitOrigin-RevId: a18ee052e9
3 years ago
Megvii Engine Team
1f0436967c
refactor(mge/distributed): using nccl as default in distributed training
GitOrigin-RevId: 81268e84bc
3 years ago
Megvii Engine Team
809d5056cd
feat(mge/distributed): enable pt shm allreduce
GitOrigin-RevId: 1dd5a02a51
4 years ago
Megvii Engine Team
13e6ea349d
feat(imperative/opr): rebase rng refactoring to dev & add python module
GitOrigin-RevId: ee5984c52d
4 years ago
Megvii Engine Team
dee5a10af9
feat(distributed): auto detect device and backend when init group
GitOrigin-RevId: 90be2d5b4d
4 years ago
Megvii Engine Team
4f3875eb4f
docs(mge): add missing docstring and fix sphinx build warnings
GitOrigin-RevId: 4ce73cfd80
4 years ago
Megvii Engine Team
364afec033
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
4 years ago
Megvii Engine Team
b9c37112a2
refactor(mge/distributed): skip barrier when running with single node
GitOrigin-RevId: 6cd5a2e1e2
4 years ago
Megvii Engine Team
b309890c66
docs(mge): pytest for sphinx docstring
GitOrigin-RevId: 8bed12562a
4 years ago
Megvii Engine Team
026af62042
docs(mge): docs typo fix
GitOrigin-RevId: 851f6de02f
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