11 Commits (8d02d10483b0c30ac31c52d6ec47cc5f1fc92c5d)

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