3 Commits (14a32ae19be59d5686026a72d49cdb5f18b0fac1)

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