Megvii Engine Team
a49d5bf973
fix(autodiff): fix inplace operation on autodiff.Function
GitOrigin-RevId: a658680f35
3 years ago
Megvii Engine Team
c28a875fac
fix(imperative/amp): adapt new transformation
GitOrigin-RevId: 6edd577a70
3 years ago
Megvii Engine Team
fc633ce4ff
fix(imperative/amp): fix custom grad in Subgraph
GitOrigin-RevId: 1c728d6ab9
3 years ago
Megvii Engine Team
c2e9860feb
chore(license): remove all license in file header
GitOrigin-RevId: a0e31247a6
3 years ago
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
0bdd0b1467
refactor(dispatch): switch to new dispatch system
GitOrigin-RevId: 32dd49a23a
3 years ago
Megvii Engine Team
7fa70df9f2
style(imperative): remove absolute imports in megengine
GitOrigin-RevId: 7004929069
3 years ago
Megvii Engine Team
c5f8b58385
docs(docstring): transfer to google style
GitOrigin-RevId: a71245c553
3 years ago
Megvii Engine Team
18274e023c
fix(imperative): fix error message when applying custom function with non-tensor arguments
GitOrigin-RevId: 387d6fda4a
3 years ago
Megvii Engine Team
522e556b83
feat(autodiff): support higher order grad
GitOrigin-RevId: 86390d2179
4 years ago
Megvii Engine Team
4f3875eb4f
docs(mge): add missing docstring and fix sphinx build warnings
GitOrigin-RevId: 4ce73cfd80
4 years ago
Megvii Engine Team
364afec033
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
4 years ago
Megvii Engine Team
2eed7d83be
docs(mge): restore autodiff.Function docs
GitOrigin-RevId: cbd84df916
4 years ago
Megvii Engine Team
a892e5d0e4
perf(mge): add more specialized grad rules
GitOrigin-RevId: f88809a6d7
4 years ago
Megvii Engine Team
4d75f691a0
feat(mge): restore remote send/recv
GitOrigin-RevId: 8b78fd5591
4 years ago
Megvii Engine Team
60c7d62a4c
refactor(imperative): remove multidispatch, raw_tensor, register
GitOrigin-RevId: ca5a6ed8eb
4 years ago
Megvii Engine Team
b5e46ae92f
feat(mge): restore Function
GitOrigin-RevId: dd455238ba
4 years ago
Megvii Engine Team
c294b9d18b
refactor(mge/tensor): remove old implementation
remove core.tensor, raw_tensor,TensorWrapper
avoid create tensor with zero-stride numpy ndarray
GitOrigin-RevId: 4fe5c4c5ba
4 years ago
Megvii Engine Team
147cef52a0
refactor(mge/imperative): implement new tensor system
GitOrigin-RevId: 2dd4e460ac
4 years ago
Megvii Engine Team
2ad8c5e1e9
fix(mge/io_remote): fix remote send/recv gradient at trace
GitOrigin-RevId: 7886efd0c1
4 years ago
Megvii Engine Team
94dba16ff4
perf(mge/imperative): misc optimizations
GitOrigin-RevId: bbe7a10b00
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
fccb2510c4
perf(mge): add opdef for elemwise and batchnorm
GitOrigin-RevId: d51fad9867
4 years ago
Megvii Engine Team
9005cf74d2
fix(mge/imperative): fix tracer leak
GitOrigin-RevId: 1a8ac20b46
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
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
Megvii Engine Team
5474538f39
refactor(mge/imperative): fork multipledispatch
GitOrigin-RevId: a7c25a4302
4 years ago
Megvii Engine Team
6b380e8965
feat(mge/imperative): run oss test and restore cmake list build items
GitOrigin-RevId: 11411b6964
4 years ago