11 Commits (ac11c38a977a935dd2230bc6a42f7158feff9c93)

Author SHA1 Message Date
  Megvii Engine Team 8d02d10483 refactor(mge/distributed): change bcast_params_ to bcast_list_ 4 years ago
  Megvii Engine Team c7acba41fc refactor(mge/optimizer): refine gradmanager api, record = __enter__ 4 years ago
  Megvii Engine Team 8c482b6709 fix(mge/grad): make register_after_backward_callback private 4 years ago
  Megvii Engine Team 66b6daf777 test(mge/optimizer): fix test for new optimizer api 4 years ago
  Megvii Engine Team e9104ef157 fix(mge/parampack): fix copy stream, import cycle 4 years ago
  Megvii Engine Team e283663a02 fix(mge/imperative): update tests to new optimizer api 4 years ago
  Megvii Engine Team b5016b9d29 feat(mge/parampack): add parampack in allreduce callback 4 years ago
  Megvii Engine Team 6070266766 refactor(mge/grad_manager): refactor gradmanager, add allreduce callback 4 years ago
  Megvii Engine Team 3f2eac2fe1 fix(mge/imperative): move functional/distributed.py to distributed/functional.py 4 years ago
  Megvii Engine Team e1fba6ece7 test(mge/distributed): add get_device_count_by_fork to fix distributed test skip 4 years ago
  Megvii Engine Team 6b380e8965 feat(mge/imperative): run oss test and restore cmake list build items 4 years ago