23 Commits (0bdd0b14670122b9f7d88847d85b2a57c438920a)

Author SHA1 Message Date
  Megvii Engine Team 0bdd0b1467 refactor(dispatch): switch to new dispatch system 3 years ago
  Megvii Engine Team ab309eb5fc feat(mgb/opr): let Split support empty IO 3 years ago
  Megvii Engine Team 207a346351 chore(mge): run get_device_count("gpu") in subprocess 4 years ago
  Megvii Engine Team 5cef74a77e feat(mge/amp): add GradScaler support 4 years ago
  Megvii Engine Team 0245594156 test(autograd): test jvp emulated by 2nd grad 3 years ago
  Megvii Engine Team 8480302da8 fix(autograd): make higher order grad experimental 4 years ago
  Megvii Engine Team 72531f2b4a test(autograd): add more tests for higher order grad 4 years ago
  Megvii Engine Team 226f24fa8e test(mge/distributed): add gather scatter reduce broadcast grad test 4 years ago
  Megvii Engine Team ff05667b48 test(dist): refactor distributed test with fixtures 4 years ago
  Megvii Engine Team 364afec033 chore(mge): update copyright years 4 years ago
  Megvii Engine Team de0742be25 refactor(mge): reopen passed assertions 4 years ago
  Megvii Engine Team 4d75f691a0 feat(mge): restore remote send/recv 4 years ago
  Megvii Engine Team 645fc6f0c8 test(mge/autodiff): add test for weird error 4 years ago
  Megvii Engine Team b5ec83c53f test(mge/autodiff): validate grad is set to None when no dependency 4 years ago
  Megvii Engine Team 9d928e7f83 refactor(mge/distributed): sync interpreter for distribtued launcher 4 years ago
  Megvii Engine Team 2ad8c5e1e9 fix(mge/io_remote): fix remote send/recv gradient at trace 4 years ago
  Megvii Engine Team 6667100638 feat(mge): use weakref for GradManger.attach 4 years ago
  Megvii Engine Team 544b798374 revert: feat(mge/grad_manager): add `clear_grad` method for GradManager 4 years ago
  Megvii Engine Team 495472954d fix(trace): link io-op to avoid deadlock 4 years ago
  Megvii Engine Team 094601e834 feat(mge/distributed): allow remote grad by using grad manager 4 years ago
  Megvii Engine Team b327822994 feat(mge/grad_manager): add `clear_grad` method for GradManager 4 years ago
  Megvii Engine Team 2627e1f7df fix(mge/grad_manager): allow multiple calls of `release` 4 years ago
  Megvii Engine Team 3af1056377 feat(mge/grad): attach grad immediately 4 years ago