Megvii Engine Team
|
c53abcdf1d
|
chore(mge): minor improvements related to grad
GitOrigin-RevId: 102467d79d
|
4 years ago |
Megvii Engine Team
|
6295f843a5
|
fix(mge/tensor): fix mem leak when getting value in trace
GitOrigin-RevId: 8bca2e7380
|
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
|
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
|
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
|
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
|
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
|
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
|
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
|
4d75f691a0
|
feat(mge): restore remote send/recv
GitOrigin-RevId: 8b78fd5591
|
4 years ago |
Megvii Engine Team
|
556e022215
|
refactor(mge/trace): remove apply on varnode
GitOrigin-RevId: 0185244c1f
|
4 years ago |
Megvii Engine Team
|
243a05b410
|
fix(mge): fix cpp trace function release
GitOrigin-RevId: 73f9642821
|
4 years ago |
Megvii Engine Team
|
9fb5581f1d
|
refactor(mge): add specialized grad rule support
GitOrigin-RevId: 141ff0a24f
|
4 years ago |
Megvii Engine Team
|
dc2507453b
|
feat(mge): add python custom op
GitOrigin-RevId: 35da0bb301
|
4 years ago |
Megvii Engine Team
|
60c44b084c
|
refactor(mge): refactor to prepare for custom grad rules
GitOrigin-RevId: 4bd8850fdf
|
4 years ago |
Megvii Engine Team
|
ae47fd4e2a
|
fix(mge): fix none return value for attrs, add test_correctness
GitOrigin-RevId: 1bb96373f4
|
4 years ago |
Megvii Engine Team
|
97d12b3e05
|
refactor(mge/imperative): fix compilation for Python 3.5
GitOrigin-RevId: edd49d0171
|
4 years ago |
Megvii Engine Team
|
87f4b46e3f
|
perf(mge/imperative): move convert_inputs from python to C++
GitOrigin-RevId: baef3d348c
|
4 years ago |
Megvii Engine Team
|
b310f2615b
|
feat(mge/imperative): implement trace and dump under new core implementation
GitOrigin-RevId: 4edc38eaf2
|
4 years ago |
Megvii Engine Team
|
15e8e7df76
|
refactor(mge/imperative): migrate to new core implementation
* swap/drop
* config/set_async_level
* _dev_tensor
* sync
GitOrigin-RevId: 850fb98852
|
4 years ago |
Megvii Engine Team
|
34c705fcbf
|
refactor(mge/imperative): move detach into C++
GitOrigin-RevId: 8c0d86cbbf
|
4 years ago |
Megvii Engine Team
|
147cef52a0
|
refactor(mge/imperative): implement new tensor system
GitOrigin-RevId: 2dd4e460ac
|
4 years ago |
Megvii Engine Team
|
577366a203
|
feat(mge/imperative): basic impl of python c extension for opdef
fix(imperative): fix refcount management on cpython opdef
refactor(mge/imperative): fix compilation for python3.6
GitOrigin-RevId: 332a516895
|
4 years ago |
Megvii Engine Team
|
9a79342d35
|
refactor(mge/virtualdep): remove virtualdep opdef
GitOrigin-RevId: bbe3ae3fa3
|
4 years ago |
Megvii Engine Team
|
88b402ef62
|
feat(mge/trace): tracing use id to set priority
GitOrigin-RevId: 6e1f1ece0e
|
4 years ago |
Megvii Engine Team
|
69e3e32240
|
feat(imperative): auto generated opdef header and python binding
GitOrigin-RevId: d2f22ad5fe
|
4 years ago |
Megvii Engine Team
|
93f4977c78
|
feat(mge/imperative): add thread name
GitOrigin-RevId: 61eece55b8
|
4 years ago |
Megvii Engine Team
|
7aa54b0ec6
|
feat(mge): enable memory swap and drop/recomputation
GitOrigin-RevId: c56c87c88b
|
4 years ago |
Megvii Engine Team
|
cc85047bf0
|
fix(mge/trace): fix sublinear in trace
GitOrigin-RevId: 356dcd9523
|
4 years ago |
Megvii Engine Team
|
ab82c8da79
|
feat(mge/device): add python binding for get_mem_status_bytes
GitOrigin-RevId: aa0d3264d6
|
4 years ago |
Megvii Engine Team
|
ab6328c563
|
feat(imperative): port persistent cache
GitOrigin-RevId: 8ca24a37cc
|
4 years ago |
Megvii Engine Team
|
798f7b3ef6
|
feat(imperative): add virtual_deps op
GitOrigin-RevId: 5845520989
|
4 years ago |
Megvii Engine Team
|
50c4daac85
|
feat(mge/interpreter): add async_level mechanism for Interpreter
GitOrigin-RevId: 8615a23b75
|
4 years ago |
Megvii Engine Team
|
1217801133
|
perf(mge): add opdef for broadcast
GitOrigin-RevId: 92f0af29eb
|
4 years ago |
Megvii Engine Team
|
fccb2510c4
|
perf(mge): add opdef for elemwise and batchnorm
GitOrigin-RevId: d51fad9867
|
4 years ago |
Megvii Engine Team
|
2c9fa7f650
|
feat(mge/profiler): add compatible and graphviz mode
GitOrigin-RevId: 40ca9ea80e
|
4 years ago |
Megvii Engine Team
|
8109cc4bd1
|
fix(imperative): set exception which worker thread throws for rendezvous
GitOrigin-RevId: f583888fdf
|
4 years ago |