9 Commits (5474538f39573f441c648f55d2dafabe69e0e450)

Author SHA1 Message Date
  Megvii Engine Team 2b3a88d011 feat(mge/distributed): add parameter replica_mode 5 years ago
  Megvii Engine Team 6d367454cf feat(mge/opr-mm): add param local_grad for collective_comm opr 5 years ago
  Megvii Engine Team 7aeb4f6ca7 fix(mge/optimizer): use static key to avoid mem leak 5 years ago
  Megvii Engine Team 01092feb9b feat(mgb): add PackAllReducePass 5 years ago
  Megvii Engine Team c7e6c658fd refactor(mge/distribute): use is_root (and rank) in stead of rank and root at collective comm 5 years ago
  Megvii Engine Team 40412e2688 feat(mge/module): add sync bn 5 years ago
  Megvii Engine Team 814a427810 feat(mge/module): add `with_parent` argument in `_flatten` 5 years ago
  Megvii Engine Team a744b3cb8a feat(mge/module):add param pack 5 years ago
  Megvii Engine Team f91881ffdc MegEngine: Initial commit of MegEngine. 5 years ago