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
12a3ef8d01
refactor(fastrun): decouple fastrun from computing graph
GitOrigin-RevId: 27abd22295
3 years ago
Megvii Engine Team
1709b3940b
perf(mge/functional): speed up Broadcast and Reshape
GitOrigin-RevId: a72f5460b6
3 years ago
Megvii Engine Team
3e206d899b
perf(mge/functional): speed up Split
GitOrigin-RevId: 43550a0706
3 years ago
Megvii Engine Team
e400b7ffe5
perf(imperative): enable memory forwarding for imperative
GitOrigin-RevId: 7c1993979c
4 years ago
Megvii Engine Team
0cb60d646d
feat(imperative): add output_descs for apply_on_physical_tensor
GitOrigin-RevId: 5b036c2c5a
3 years ago
Megvii Engine Team
c55fda9a7c
fix(fastrun): don't kill profiling worker
GitOrigin-RevId: 99a0f11a5a
4 years ago
Megvii Engine Team
aa587446fc
feat(subgraph): support shape inference for CompiledOp
GitOrigin-RevId: a96b8f3446
4 years ago
Megvii Engine Team
95ac055538
feat(dnn,mgb,imperative): add diag opr implement
GitOrigin-RevId: 43016ffa2b
4 years ago
Megvii Engine Team
cbbca5fb10
feat(mge): add softmax op use cudnn api
GitOrigin-RevId: 7734ebf8c4
4 years ago
Megvii Engine Team
2faa6ea5a9
Merge pull request #213 from kxz18:rnn
GitOrigin-RevId: 9e9215c115
4 years ago
Megvii Engine Team
4b0ecb5deb
fix(ops/recv): use std::vector to store shape to support scalar
GitOrigin-RevId: e1dac3c919
4 years ago
Megvii Engine Team
1999307015
feat(mgb/opr): add dropout kernel
GitOrigin-RevId: d248bd2005
4 years ago
Megvii Engine Team
a93741815b
feat(mgb/opr): add layernorm forward and backward kernel
GitOrigin-RevId: 0cd484e753
4 years ago
Megvii Engine Team
6011f51001
style(all): fix clang-format for MGB_DEFINE inside another macro
GitOrigin-RevId: 8c2b6a2aed
4 years ago
Megvii Engine Team
7231257efc
fix(imperative/fastrun): fix worksapce limit for cpu compnode
GitOrigin-RevId: 4583ce6d4b
4 years ago
Megvii Engine Team
8624ec224b
fix(mgb): fix param merge bug that caused the weight statistics error
GitOrigin-RevId: f76a096832
4 years ago
Megvii Engine Team
cf1db2616e
fix(fastrun): replace py_redis with cpp_redis to avoid deadlock
GitOrigin-RevId: 9af7fa5c97
4 years ago
Megvii Engine Team
ce119ef5a5
fix(lite): fix lite error when record level is 2
GitOrigin-RevId: 7dabfd8876
4 years ago
kxz@thumt102-1
8f48da7ffe
feat(mgb/opr): add cell level rnn/lstm and sequence level rnn/lstm
4 years ago
Megvii Engine Team
2881934cb8
feat(dnn/check_non_finite): addmul scale to check_non_finite opr
GitOrigin-RevId: c35a219e52
4 years ago
Megvii Engine Team
b8ccc6a211
fix(mgb): fix loss execution policy after opr shallow copy
GitOrigin-RevId: 4738136e4a
4 years ago
Megvii Engine Team
501eadc1db
fix(mgb): fix copybara mc20
GitOrigin-RevId: 2b491e2278
4 years ago
Megvii Engine Team
2b8e7940b6
fix(lite/cambricon): fix cambricon models which have multiple comp node
GitOrigin-RevId: 624fd7f0ce
4 years ago
Megvii Engine Team
0c3699ff55
test(imperative/python): fix testcase for magicmind runtime module
GitOrigin-RevId: baf2f72f01
4 years ago
Megvii Engine Team
f398c8e619
feat(imperative): add magicmind runtime opr
GitOrigin-RevId: 02ddb886d9
4 years ago
Megvii Engine Team
b5284796f9
feat(mgb/cambricon): add magicmind runtime opr
GitOrigin-RevId: 51af7508de
4 years ago
Megvii Engine Team
623477fdeb
refactor(megbrain): combine mc40 and mc20 compnode and change mc20 to ax
GitOrigin-RevId: bcef8c6f04
4 years ago
Megvii Engine Team
b8f810ee31
feat(megbrain): add mc20 runtime opr
GitOrigin-RevId: 2ee07b213b
4 years ago
Megvii Engine Team
67e4e83419
fix(lite): fix the force_output_use_user_specified_memory when out var not supported
GitOrigin-RevId: ffaf4c1416
4 years ago
Megvii Engine Team
24f12df95b
chore(release): bump version
GitOrigin-RevId: 6903f9a962
4 years ago
Megvii Engine Team
0be6ca8899
fix(src/core): fix record change ptr bug on comp node copy
GitOrigin-RevId: 0f68966211
4 years ago
Megvii Engine Team
b59e8ccf24
fix(mgb): fix cambricon bangc copybara
GitOrigin-RevId: 22b57364e8
4 years ago
Megvii Engine Team
fe6299c52d
Revert "feat(mgb): allow output tensor's ptr change when record"
This reverts commit c610c8bf9a .
GitOrigin-RevId: 2eb0504c5d
4 years ago
Megvii Engine Team
d90cb7763c
feat(src/core): record support change ptr basic
GitOrigin-RevId: b12f1c4a66
4 years ago
Megvii Engine Team
0a56a5b110
feat(cuda/comp_node): gets (maximum) GPU memory allocated/reserved
GitOrigin-RevId: da2cc22436
4 years ago
Megvii Engine Team
13c7c572df
fix(mgb): fix shape infer's condition in lite
GitOrigin-RevId: 550eaff4cd
4 years ago
Megvii Engine Team
0ddabb06cc
fix(imperative/astype): fix astype when target dtype is int4
GitOrigin-RevId: 1c5195176d
4 years ago
Megvii Engine Team
481007812e
fix(mgb): support rc version
GitOrigin-RevId: aa5fe5198d
4 years ago
Megvii Engine Team
8b7649345a
feat(lite): lite support output var no copy option
GitOrigin-RevId: 5b9488cb93
4 years ago
Megvii Engine Team
7642f66df7
feat(core): support output var no copy
GitOrigin-RevId: fd68a1735b
4 years ago
Megvii Engine Team
7fa5f6f4e2
fix(mgb): correct version of fbs serialization
GitOrigin-RevId: 2a71d9afc2
4 years ago
Megvii Engine Team
25ec2530ba
feat(whl/api/lar): enable megengine dll on Windows
1: reduce python whl package size
2: unify api link logic on all OS
3: add option: MGE_WINDOWS_BUILD_WITH_STATIC_CRT
--- default OFF
--- if build CRT(vc runtime) with STATIC with megengine.dll
some CRT api will crash, for example, flush, so if you
build with static megengine, and do not want to install CRT
you can set MGE_WINDOWS_BUILD_WITH_STATIC_CRT TRUE
--- how to install CRT:
https://docs.microsoft.com/en-us/cpp/windows/latest-supported-vc-redist?view=msvc-160
install VC_redist.x64.exe
4: rename megengine_export to megengine_shared(only export needed symbols ),
caused by runtime symbols conflict with pytorch
GitOrigin-RevId: 93d8d80f29
4 years ago
Megvii Engine Team
c17d603847
feat(mgb): allow output tensor's ptr change when record
GitOrigin-RevId: c610c8bf9a
4 years ago
Megvii Engine Team
d9a9d9d49e
fix(imperative/fastrun): set workspace limit for imperative rt
GitOrigin-RevId: 474dc691a3
4 years ago
Megvii Engine Team
bd42cb06cf
refactor(mgb/lite): refactor lite InfilePersistentCache with core impl
GitOrigin-RevId: 64b7825c34
4 years ago
Megvii Engine Team
a4ac5e7e8f
fix(mgb/gopt): fix profiler cache when build without opr type info name
GitOrigin-RevId: 0eba678f2b
4 years ago
Megvii Engine Team
012de7695d
feat(mgb/gopt): add profiler cache
In order to improve performance of the profiling procedure. Make layout transform testcase stable. The profiling result in ci environment will be cached in files.
GitOrigin-RevId: ba2743f35f
4 years ago
Megvii Engine Team
565466c25f
feat(lite): auto deduce output tensor shape before model forward
GitOrigin-RevId: 78e00dab5d
4 years ago
Megvii Engine Team
369c2ccc5a
style(all): reformat c++ code
GitOrigin-RevId: 3ffd1b211f
4 years ago