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 |
Megvii Engine Team
|
2e9ba679b5
|
feat(mge/device): __repr__ method will show physical device
GitOrigin-RevId: 050c3864a7
|
4 years ago |
Megvii Engine Team
|
3246ee5e87
|
perf(mge): use DepType::HOST_VALUE in trace when possible
GitOrigin-RevId: 5d47ed263f
|
4 years ago |
Megvii Engine Team
|
be2366423d
|
feat(sdk/load_and_run): add output-strip-info for dump with testcase imperative
GitOrigin-RevId: 337d95c7c2
|
4 years ago |
Megvii Engine Team
|
36a4fb5611
|
ci(image): update docker image install python package
GitOrigin-RevId: 171e95b3d9
|
4 years ago |
Megvii Engine Team
|
53ec6b833e
|
fix(mge): fix graph binding refcnt
GitOrigin-RevId: c955eb260a
|
4 years ago |
Megvii Engine Team
|
296a288594
|
fix(build/macos): fix macos build
GitOrigin-RevId: 66ec91c647
|
4 years ago |
Megvii Engine Team
|
fc74d5f2c3
|
feat(mge/dump): add more options for megbrain_graph.dump_graph
GitOrigin-RevId: 848e3c87c1
|
4 years ago |
Megvii Engine Team
|
a7b9ece4a7
|
feat(mgb/comp_node): add set_prealloc_config
GitOrigin-RevId: e725e7efdd
|
4 years ago |
Megvii Engine Team
|
91efd67dbf
|
refactor(mge/jit): change dump options, add test
GitOrigin-RevId: fbc0d51c2b
|
4 years ago |
Megvii Engine Team
|
ac11c38a97
|
feat(mge/imperative): add graph load and cgtools for imperative
GitOrigin-RevId: ba251f452a
|
4 years ago |
Megvii Engine Team
|
6f581906a1
|
refactor(mge/profiler): refactor profiler
GitOrigin-RevId: 279aa779a6
|
4 years ago |
Megvii Engine Team
|
5474b000a7
|
fix(mge/functional): fix convert_inputs before apply
GitOrigin-RevId: ab41974a1f
|
4 years ago |
Megvii Engine Team
|
478f2c51dc
|
feat(mge/jit): add trace/dump options
GitOrigin-RevId: 0f43c14599
|
4 years ago |
Megvii Engine Team
|
b1ab3646f5
|
feat(imperative): add tensor sanity check
GitOrigin-RevId: 27243978e3
|
4 years ago |
Megvii Engine Team
|
fc3eca845b
|
feat(mge/imperative/jit): add dump input shape and xornet example
GitOrigin-RevId: 5e2acd4052
|
4 years ago |
Megvii Engine Team
|
38a5c1c970
|
fix(mge): fix refcnt in TensorShape caster
GitOrigin-RevId: 24f6c7a6d0
|
4 years ago |
Megvii Engine Team
|
2cc85487d1
|
fix(imperative): fix hardcode of default device
GitOrigin-RevId: 722c4debfa
|
4 years ago |
Megvii Engine Team
|
403a1e7b24
|
feat(mge/imperative): add graph profiler
GitOrigin-RevId: c73563f337
|
4 years ago |
Megvii Engine Team
|
0bb4969696
|
fix(imperative): fix dangling pointer in dispatcher
GitOrigin-RevId: b21f7e06f0
|
4 years ago |
Megvii Engine Team
|
799d09ef13
|
fix(imperative): fix compile error when vectorcall is enabled
GitOrigin-RevId: 7dda34a758
|
4 years ago |