Megvii Engine Team
|
e0fffa5286
|
fix(mge/tensor): fix mem leak when getting value in trace
GitOrigin-RevId: 1dabd88e09
|
4 years ago |
Megvii Engine Team
|
3a4e0598ba
|
Revert "perf(mge): add memory optimization for backward graph"
This reverts commit d67043ba82 .
GitOrigin-RevId: 2be2e2ef16
|
4 years ago |
Megvii Engine Team
|
5f299dd08e
|
chore(release): bump version
|
4 years ago |
Megvii Engine Team
|
27503620a5
|
fix(fastrun): release free blocks before profiling
GitOrigin-RevId: b0d9887e9a
|
4 years ago |
Megvii Engine Team
|
08cc10324e
|
fix(imperative): fix persistent_cache
GitOrigin-RevId: 8f7bb5899f
|
4 years ago |
Megvii Engine Team
|
998f71a8c9
|
perf(mge/imperative): do not busy wait in imperative's queue
GitOrigin-RevId: de8db51096
|
4 years ago |
Megvii Engine Team
|
62637fc46e
|
refactor(mgb/utils): allow user set different spin time for ASyncSCQueue
GitOrigin-RevId: 1c6c4a7a16
|
4 years ago |
Megvii Engine Team
|
d07cfdcbe9
|
refactor(mgb): move convolution mixin to search_policy
GitOrigin-RevId: 81e32da034
|
4 years ago |
Megvii Engine Team
|
a57317f4e1
|
refactor(mgb): move profile cache out of mgb opr and update CACHE_KEY_VERSION
GitOrigin-RevId: db433164d2
|
4 years ago |
Megvii Engine Team
|
278b2baa8c
|
perf(mge): add memory optimization for backward graph
precompute ops in forward to reduce saved tensor size
GitOrigin-RevId: d67043ba82
|
4 years ago |
Megvii Engine Team
|
ebe8689267
|
feat(sdk/load_and_run): add flags --io-dump-stdout and --io-dump-stderr
GitOrigin-RevId: 64d572bdfc
|
4 years ago |
Megvii Engine Team
|
5c7d48cdb9
|
fix(mge/functional): fix tensor split
GitOrigin-RevId: 0a112ab0bd
|
4 years ago |
Megvii Engine Team
|
a240d558f1
|
fix(mge/dnn): fix rng and topk oom in distributed training
GitOrigin-RevId: 9841d1219e
|
4 years ago |
Megvii Engine Team
|
c3c1e46df7
|
chore(mgblar): delete winograd transform switch
GitOrigin-RevId: aa4051541d
|
4 years ago |
Megvii Engine Team
|
82847bc70a
|
docs(mge): add conv2d output shape computation formula
GitOrigin-RevId: 7ee476a80d
|
4 years ago |
Megvii Engine Team
|
d3bfb0e983
|
fix(mge): fix trace exit code and reformat
GitOrigin-RevId: 145c06b7e7
|
4 years ago |
Megvii Engine Team
|
23b9a98f5e
|
fix(mge): fix sublnear cuda and mem leak
GitOrigin-RevId: 82091ec9a6
|
4 years ago |
Megvii Engine Team
|
c70a49ed2c
|
fix(mge): correct trace outputs when grad does copy
GitOrigin-RevId: 65c8956a7d
|
4 years ago |
Megvii Engine Team
|
d4ada69d3b
|
refactor(mge): trace exception in compiled info
GitOrigin-RevId: 508f5463b9
|
4 years ago |
Megvii Engine Team
|
c9c3429a6b
|
refactor(mge): fix sublinear
GitOrigin-RevId: 5bb0383781
|
4 years ago |
Megvii Engine Team
|
de0742be25
|
refactor(mge): reopen passed assertions
GitOrigin-RevId: e0276e73e3
|
4 years ago |
Megvii Engine Team
|
a90c937d22
|
feat(interpreter): add command buffer for inplace
GitOrigin-RevId: 020d1e88d4
|
4 years ago |
Megvii Engine Team
|
09af925fb6
|
fix(mge): fix cpp trace function release
GitOrigin-RevId: 924f945c21
|
4 years ago |
Megvii Engine Team
|
3975a54aa4
|
fix(mge/sublinear): fix divide by zero in sublinear
GitOrigin-RevId: d2bd113809
|
4 years ago |
Megvii Engine Team
|
8f7f52ae4d
|
feat(jit): add memfwd in jit executor opr
GitOrigin-RevId: b58860bbe8
|
4 years ago |
Megvii Engine Team
|
dfb2b2ce49
|
fix(dnn): change pooling window size smaller than padding constraint to log_error
GitOrigin-RevId: c3cda68f6d
|
4 years ago |
Megvii Engine Team
|
6919127fd4
|
fix(mge): fix event pool destructor order issue and do full sync at python exit
GitOrigin-RevId: b9fa1f45b1
|
4 years ago |
Megvii Engine Team
|
60459bd79b
|
fix(mge/pytest): remove sync in pytest teardown & add full_sync
GitOrigin-RevId: ee278a8ac1
|
4 years ago |
Megvii Engine Team
|
c49427d15a
|
feat(imperative): add inplace add_update option in optimizer
GitOrigin-RevId: b8feb49321
|
4 years ago |
Megvii Engine Team
|
aeb7980b29
|
perf(mgb): outputs of the same opr and same compnode share the same callbackcaller
GitOrigin-RevId: 59b8e3bcbe
|
4 years ago |
Megvii Engine Team
|
89b6dbc74c
|
fix(mge/imperative): import subpackages
GitOrigin-RevId: 342f6c87cc
|
4 years ago |
Megvii Engine Team
|
7f3f9a94ae
|
feat(mgb/core): add shape hint for graph optimization
GitOrigin-RevId: eaad25a7ef
|
4 years ago |
Megvii Engine Team
|
d1fbec4fe2
|
feat(dnn/atlas): add atlas stub
GitOrigin-RevId: c63294378e
|
4 years ago |
Megvii Engine Team
|
a85531dd0f
|
feat(mgb/opr): add tqt opr
GitOrigin-RevId: 49c62cd532
|
4 years ago |
Megvii Engine Team
|
44742e32e9
|
fix(mge/api): check input dim of dot and mark output as scalar
GitOrigin-RevId: a3ba7e099c
|
4 years ago |
Megvii Engine Team
|
697f70c011
|
feat(mge/pytest): add more tests for specialized grad rules
GitOrigin-RevId: 509ef5a220
|
4 years ago |
Megvii Engine Team
|
cf3f58cb9b
|
fix(mge/autodiff): fix segfault when grad is nullptr
GitOrigin-RevId: 6139212bfd
|
4 years ago |
Megvii Engine Team
|
288c2e08b5
|
fix(mge/autodiff): fix expand_dims and grad rule fallback
GitOrigin-RevId: 4aae771222
|
4 years ago |
Megvii Engine Team
|
a5609f3bbf
|
fix(cambricon): fix cross cn copy for cambricon
GitOrigin-RevId: 21942a82a3
|
4 years ago |
Megvii Engine Team
|
05c739b846
|
refactor(mge/data): rename `MapDataset` to `Dataset`
GitOrigin-RevId: 6262561355
|
4 years ago |
Megvii Engine Team
|
a892e5d0e4
|
perf(mge): add more specialized grad rules
GitOrigin-RevId: f88809a6d7
|
4 years ago |
Megvii Engine Team
|
e9e5f442a7
|
fix(mge): expand custom op before trace
GitOrigin-RevId: 725a5b87cb
|
4 years ago |
Megvii Engine Team
|
3faba54f28
|
fix(mge): fix segfault with Function returning unused grads
GitOrigin-RevId: 0cce845923
|
4 years ago |
Megvii Engine Team
|
75129cf071
|
chore(mge): clean up before merge to dev
* remove dead test
* clean some codes
* fix test_fake_quant due to change of implementation
GitOrigin-RevId: f030a9966d
|
4 years ago |
Megvii Engine Team
|
aba0acc797
|
fix(sdk): add AssertEqual opr, fix dump_with_testcase_mge
GitOrigin-RevId: 6f797570b6
|
4 years ago |
Megvii Engine Team
|
dd9f54cdfa
|
refactor(mge): migrate to new core implementation
* remove dispatcher/interpreter python wrapper
* rename tensor_wrapper to array_method
GitOrigin-RevId: b8a402c2be
|
4 years ago |
Megvii Engine Team
|
b9762d714c
|
fix(mge): make parampack run with tensor symbolic shape
GitOrigin-RevId: 6fc313785d
|
4 years ago |
Megvii Engine Team
|
4d75f691a0
|
feat(mge): restore remote send/recv
GitOrigin-RevId: 8b78fd5591
|
4 years ago |
Megvii Engine Team
|
9c92701f63
|
feat(mge): support python -m megengine.distributed.server
GitOrigin-RevId: f1e5c8e3cf
|
4 years ago |
Megvii Engine Team
|
556e022215
|
refactor(mge/trace): remove apply on varnode
GitOrigin-RevId: 0185244c1f
|
4 years ago |