Megvii Engine Team
194201a774
revert: chore(mge/misc): api converage
This reverts commit --
77fd432cb2
0d7126641b
559d205a8c
6d8e7398e4
b72517bfe9
GitOrigin-RevId: 18dc4ce999
2 years ago
Megvii Engine Team
98460f583f
chore(mge/misc): rename APIs that should not be public
GitOrigin-RevId: b72517bfe9
2 years ago
Megvii Engine Team
d313f92610
fix(imperative/amp): fix format transformation for symbol trans
GitOrigin-RevId: 96cc237c67
3 years ago
Megvii Engine Team
e393d1cf65
feat(mge/amp): add convert_format module for NHWC training
GitOrigin-RevId: 1b41e1042c
3 years ago
Megvii Engine Team
c2e9860feb
chore(license): remove all license in file header
GitOrigin-RevId: a0e31247a6
3 years ago
Megvii Engine Team
87f00232f2
fix(mge/gm): fix missing dtype checking while attach tensors
GitOrigin-RevId: f0aaea99b9
3 years ago
Megvii Engine Team
0bdd0b1467
refactor(dispatch): switch to new dispatch system
GitOrigin-RevId: 32dd49a23a
3 years ago
Megvii Engine Team
c5f8b58385
docs(docstring): transfer to google style
GitOrigin-RevId: a71245c553
3 years ago
Megvii Engine Team
5cef74a77e
feat(mge/amp): add GradScaler support
GitOrigin-RevId: 0ab4910360
4 years ago
Megvii Engine Team
72531f2b4a
test(autograd): add more tests for higher order grad
GitOrigin-RevId: 5fc308f87a
4 years ago
Megvii Engine Team
522e556b83
feat(autodiff): support higher order grad
GitOrigin-RevId: 86390d2179
4 years ago
Megvii Engine Team
01354337a9
fix(mge/autodiff): fix incorrect handling of tuple dy
GitOrigin-RevId: beca8e3711
4 years ago
Megvii Engine Team
efc7f04fd3
refactor(mge/jit): using static global_enable for apply ctx insted of global variable
GitOrigin-RevId: dd82b53faf
4 years ago
Megvii Engine Team
696d2c2ef1
fix(mge/autodiff): check tensors to be attached
GitOrigin-RevId: c4f3f80876
4 years ago
Megvii Engine Team
fe99cdc794
feat(interpreter): add dynamic sublinear
GitOrigin-RevId: 7de54fe743
4 years ago
Megvii Engine Team
fa4bf16800
feat(mge/functional): add repeat and tile opr
GitOrigin-RevId: a20d4b6fb0
4 years ago
Megvii Engine Team
30d6b4f65d
fix(mge): fix scalar parameter change to 1-dim parameter after backward and optimize
GitOrigin-RevId: 1794369a71
4 years ago
Megvii Engine Team
dbb3dd681f
refactor(profiler): integrate profiler into interpreter
GitOrigin-RevId: ccc984acbd
4 years ago
Megvii Engine Team
364afec033
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
4 years ago
Megvii Engine Team
5100317660
feat(mge): make Function available at autodiff.Function
GitOrigin-RevId: 7c9ab65351
4 years ago
Megvii Engine Team
147cef52a0
refactor(mge/imperative): implement new tensor system
GitOrigin-RevId: 2dd4e460ac
4 years ago
Megvii Engine Team
81e0dae4ac
docs(mge): update GradManger docs
GitOrigin-RevId: ccffb830c4
4 years ago
Megvii Engine Team
6667100638
feat(mge): use weakref for GradManger.attach
GitOrigin-RevId: 6df336c3c1
4 years ago
Megvii Engine Team
75ca5bfe65
feat(mge): remove GradManager.detach until it is ready
GitOrigin-RevId: 1302b25b2a
4 years ago
Megvii Engine Team
544b798374
revert: feat(mge/grad_manager): add `clear_grad` method for GradManager
This reverts commit aa9540e090
.
GitOrigin-RevId: 2d15e503de
4 years ago
Megvii Engine Team
094601e834
feat(mge/distributed): allow remote grad by using grad manager
GitOrigin-RevId: a890c206a5
4 years ago
Megvii Engine Team
b309890c66
docs(mge): pytest for sphinx docstring
GitOrigin-RevId: 8bed12562a
4 years ago
Megvii Engine Team
b327822994
feat(mge/grad_manager): add `clear_grad` method for GradManager
GitOrigin-RevId: aa9540e090
4 years ago
Megvii Engine Team
2627e1f7df
fix(mge/grad_manager): allow multiple calls of `release`
GitOrigin-RevId: 38ca4c78ff
4 years ago
Megvii Engine Team
3af1056377
feat(mge/grad): attach grad immediately
GitOrigin-RevId: e3a168c03a
4 years ago
Megvii Engine Team
6701346398
docs(mge): fix some docstring format problem
GitOrigin-RevId: cbc5ab04b3
4 years ago
Megvii Engine Team
0d4568d6f8
feat(mge): rename broadcast -> broadcast_to
GitOrigin-RevId: 82f46ad2c2
4 years ago
Megvii Engine Team
026af62042
docs(mge): docs typo fix
GitOrigin-RevId: 851f6de02f
4 years ago
Megvii Engine Team
e6715910c5
docs(mge/imperative): add docstring for GradManager
GitOrigin-RevId: 4c326206b8
4 years ago
Megvii Engine Team
10e942d952
refactor(mge): polish api
- refactor(mge): add support for optimizer.step().clear_grad() idiom
- refactor(mge): rename some methods of GradManager
- refactor(mge): remove tensor_nn and TensorDict
- refactor(mge): remove Buffer
- refactor(mge): remove requires_grad flag
- refactor(mge): add a default grad=None attribute to Tensor
- refactor(mge): deprecation for 1.0
GitOrigin-RevId: 3b723d9387
4 years ago
Megvii Engine Team
c7acba41fc
refactor(mge/optimizer): refine gradmanager api, record = __enter__
GitOrigin-RevId: 5376177237
4 years ago
Megvii Engine Team
8c482b6709
fix(mge/grad): make register_after_backward_callback private
GitOrigin-RevId: 8eb6c0e628
4 years ago
Megvii Engine Team
e9104ef157
fix(mge/parampack): fix copy stream, import cycle
GitOrigin-RevId: 673e11c5b6
4 years ago
Megvii Engine Team
e283663a02
fix(mge/imperative): update tests to new optimizer api
GitOrigin-RevId: 3d06e3db3c
4 years ago
Megvii Engine Team
b5016b9d29
feat(mge/parampack): add parampack in allreduce callback
GitOrigin-RevId: 73d53eeba1
4 years ago
Megvii Engine Team
5ae89c799b
refactor(mgb/grad): place grad at param.grad
GitOrigin-RevId: fddeace402
4 years ago
Megvii Engine Team
9faa32fcb7
fix(mge/imperative): fix grad callback
GitOrigin-RevId: 6f843b0106
4 years ago
Megvii Engine Team
6070266766
refactor(mge/grad_manager): refactor gradmanager, add allreduce callback
GitOrigin-RevId: 086e2871e8
4 years ago