Megvii Engine Team
273c0e8745
fix(autodiff): fix some bugs in relation to 2nd order grad
1. implement double backward for batchnorm
2. fix grad attach in nested grad manager
3. pad empty tensor for unsatisfied output_has_grad
4. support double backward for jit subgraph
5. support double backward for autodiff.Function
6. readd debug flag MGE_LOG_OP_DISPATCH
GitOrigin-RevId: cd31ddc620
3 years ago
Megvii Engine Team
177001d5e5
refactor(dispatch): allow dynamic type creation
GitOrigin-RevId: 27dde05cff
3 years ago
Megvii Engine Team
0bdd0b1467
refactor(dispatch): switch to new dispatch system
GitOrigin-RevId: 32dd49a23a
4 years ago
Megvii Engine Team
369c2ccc5a
style(all): reformat c++ code
GitOrigin-RevId: 3ffd1b211f
4 years ago
Megvii Engine Team
000517c641
fix(grad): stop using exception in grad_override
GitOrigin-RevId: 00ae38d48b
4 years ago
Megvii Engine Team
8480302da8
fix(autograd): make higher order grad experimental
GitOrigin-RevId: 81e1eb0ebf
4 years ago
Megvii Engine Team
522e556b83
feat(autodiff): support higher order grad
GitOrigin-RevId: 86390d2179
4 years ago
Megvii Engine Team
4f240ec2d3
refactor(mge/jit): make trace return any kind of output
GitOrigin-RevId: fd1265c661
4 years ago
Megvii Engine Team
364afec033
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
5 years ago
Megvii Engine Team
4d75f691a0
feat(mge): restore remote send/recv
GitOrigin-RevId: 8b78fd5591
5 years ago
Megvii Engine Team
9fb5581f1d
refactor(mge): add specialized grad rule support
GitOrigin-RevId: 141ff0a24f
5 years ago
Megvii Engine Team
147cef52a0
refactor(mge/imperative): implement new tensor system
GitOrigin-RevId: 2dd4e460ac
5 years ago