22 Commits (43cb305e53f87f69c1039c06829aa00a3a7f3f66)

Author SHA1 Message Date
  Megvii Engine Team ab309eb5fc feat(mgb/opr): let Split support empty IO 4 years ago
  Megvii Engine Team 207a346351 chore(mge): run get_device_count("gpu") in subprocess 5 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 4 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 5 years ago
  Megvii Engine Team 364afec033 chore(mge): update copyright years 5 years ago
  Megvii Engine Team de0742be25 refactor(mge): reopen passed assertions 5 years ago
  Megvii Engine Team 4d75f691a0 feat(mge): restore remote send/recv 5 years ago
  Megvii Engine Team 645fc6f0c8 test(mge/autodiff): add test for weird error 5 years ago
  Megvii Engine Team b5ec83c53f test(mge/autodiff): validate grad is set to None when no dependency 5 years ago
  Megvii Engine Team 9d928e7f83 refactor(mge/distributed): sync interpreter for distribtued launcher 5 years ago
  Megvii Engine Team 2ad8c5e1e9 fix(mge/io_remote): fix remote send/recv gradient at trace 5 years ago
  Megvii Engine Team 6667100638 feat(mge): use weakref for GradManger.attach 5 years ago
  Megvii Engine Team 544b798374 revert: feat(mge/grad_manager): add `clear_grad` method for GradManager 5 years ago
  Megvii Engine Team 495472954d fix(trace): link io-op to avoid deadlock 5 years ago
  Megvii Engine Team 094601e834 feat(mge/distributed): allow remote grad by using grad manager 5 years ago
  Megvii Engine Team b327822994 feat(mge/grad_manager): add `clear_grad` method for GradManager 5 years ago
  Megvii Engine Team 2627e1f7df fix(mge/grad_manager): allow multiple calls of `release` 5 years ago
  Megvii Engine Team 3af1056377 feat(mge/grad): attach grad immediately 5 years ago