37 Commits (0bdd0b14670122b9f7d88847d85b2a57c438920a)

Author SHA1 Message Date
  Megvii Engine Team 0bdd0b1467 refactor(dispatch): switch to new dispatch system 3 years ago
  Megvii Engine Team c5f8b58385 docs(docstring): transfer to google style 3 years ago
  Megvii Engine Team 5cef74a77e feat(mge/amp): add GradScaler support 4 years ago
  Megvii Engine Team 72531f2b4a test(autograd): add more tests for higher order grad 4 years ago
  Megvii Engine Team 522e556b83 feat(autodiff): support higher order grad 4 years ago
  Megvii Engine Team 01354337a9 fix(mge/autodiff): fix incorrect handling of tuple dy 4 years ago
  Megvii Engine Team efc7f04fd3 refactor(mge/jit): using static global_enable for apply ctx insted of global variable 4 years ago
  Megvii Engine Team 696d2c2ef1 fix(mge/autodiff): check tensors to be attached 4 years ago
  Megvii Engine Team fe99cdc794 feat(interpreter): add dynamic sublinear 4 years ago
  Megvii Engine Team fa4bf16800 feat(mge/functional): add repeat and tile opr 4 years ago
  Megvii Engine Team 30d6b4f65d fix(mge): fix scalar parameter change to 1-dim parameter after backward and optimize 4 years ago
  Megvii Engine Team dbb3dd681f refactor(profiler): integrate profiler into interpreter 4 years ago
  Megvii Engine Team 364afec033 chore(mge): update copyright years 4 years ago
  Megvii Engine Team 5100317660 feat(mge): make Function available at autodiff.Function 4 years ago
  Megvii Engine Team 147cef52a0 refactor(mge/imperative): implement new tensor system 4 years ago
  Megvii Engine Team 81e0dae4ac docs(mge): update GradManger docs 4 years ago
  Megvii Engine Team 6667100638 feat(mge): use weakref for GradManger.attach 4 years ago
  Megvii Engine Team 75ca5bfe65 feat(mge): remove GradManager.detach until it is ready 4 years ago
  Megvii Engine Team 544b798374 revert: feat(mge/grad_manager): add `clear_grad` method for GradManager 4 years ago
  Megvii Engine Team 094601e834 feat(mge/distributed): allow remote grad by using grad manager 4 years ago
  Megvii Engine Team b309890c66 docs(mge): pytest for sphinx docstring 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
  Megvii Engine Team 6701346398 docs(mge): fix some docstring format problem 4 years ago
  Megvii Engine Team 0d4568d6f8 feat(mge): rename broadcast -> broadcast_to 4 years ago
  Megvii Engine Team 026af62042 docs(mge): docs typo fix 4 years ago
  Megvii Engine Team e6715910c5 docs(mge/imperative): add docstring for GradManager 4 years ago
  Megvii Engine Team 10e942d952 refactor(mge): polish api 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 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 5ae89c799b refactor(mgb/grad): place grad at param.grad 4 years ago
  Megvii Engine Team 9faa32fcb7 fix(mge/imperative): fix grad callback 4 years ago
  Megvii Engine Team 6070266766 refactor(mge/grad_manager): refactor gradmanager, add allreduce callback 4 years ago