3 Commits (f856b1708592c5f80320659654f8159c9a38c088)

Author SHA1 Message Date
  Megvii Engine Team 6d367454cf feat(mge/opr-mm): add param local_grad for collective_comm opr 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