Megvii Engine Team
b310f2615b
feat(mge/imperative): implement trace and dump under new core implementation
GitOrigin-RevId: 4edc38eaf2
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
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
7aa54b0ec6
feat(mge): enable memory swap and drop/recomputation
GitOrigin-RevId: c56c87c88b
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
638ab52fdc
feat(mge/imperative): simulates scalar
GitOrigin-RevId: e81630e256
4 years ago
Megvii Engine Team
cc85047bf0
fix(mge/trace): fix sublinear in trace
GitOrigin-RevId: 356dcd9523
4 years ago
Megvii Engine Team
495472954d
fix(trace): link io-op to avoid deadlock
GitOrigin-RevId: 872cb6b715
4 years ago
Megvii Engine Team
2e53077915
fix(mge/trace): use xpux device when dump
GitOrigin-RevId: f37285f70e
4 years ago
Megvii Engine Team
7cd846a570
fix(mge/batch_norm): fix batch_norm check when trace(symbolic=True)
GitOrigin-RevId: 2032eb5f7d
4 years ago
Megvii Engine Team
b309890c66
docs(mge): pytest for sphinx docstring
GitOrigin-RevId: 8bed12562a
4 years ago
Megvii Engine Team
1b56851715
fix(mge/trace): fix graph option in trace
GitOrigin-RevId: 7bec84f56d
4 years ago
Megvii Engine Team
9748aebeea
refactor(mge): tensor_shape -> symbolic_shape
GitOrigin-RevId: 366dc048bf
4 years ago
Megvii Engine Team
af349d6102
fix(mge/functional): fix op mismatch when tracing NMSKeep
GitOrigin-RevId: e8f2cbb755
4 years ago
Megvii Engine Team
7ac4dbc27d
fix(mgb/trace): finalize when exception raise
GitOrigin-RevId: b8ffd00a7e
4 years ago
Megvii Engine Team
b2f9efb897
perf(mge): always async dispatch in jit.trace
GitOrigin-RevId: 6fa80b0dd9
4 years ago
Megvii Engine Team
59a9275c66
fix(mge): fix optimize_for_inference during trace.dump
GitOrigin-RevId: e10f7c323a
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
4ab259f558
refactor(mge): add some deprecation and migration guides
GitOrigin-RevId: 4793adf02b
4 years ago
Megvii Engine Team
91efd67dbf
refactor(mge/jit): change dump options, add test
GitOrigin-RevId: fbc0d51c2b
4 years ago
Megvii Engine Team
39d4328309
fix(mge/functional): functional api fixes
GitOrigin-RevId: fa206c4ff6
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
76f367962f
fix(mge/trace): fix op order in symbolic
GitOrigin-RevId: fbf081a199
4 years ago
Megvii Engine Team
10e942d952
refactor(mge): polish api
- refactor(mge): add support for optimizer.step().clear_grad() idiom
- refactor(mge): rename some methods of GradManager
- refactor(mge): remove tensor_nn and TensorDict
- refactor(mge): remove Buffer
- refactor(mge): remove requires_grad flag
- refactor(mge): add a default grad=None attribute to Tensor
- refactor(mge): deprecation for 1.0
GitOrigin-RevId: 3b723d9387
4 years ago
Megvii Engine Team
478f2c51dc
feat(mge/jit): add trace/dump options
GitOrigin-RevId: 0f43c14599
4 years ago
Megvii Engine Team
d49a5c1be2
fix(mge/imperative): remove codes newly introduced in numpy 1.19
GitOrigin-RevId: 874350cbc7
4 years ago
Megvii Engine Team
a66d4b8bb8
fix(mge/parampacksplit): fix param pack split mem forward
GitOrigin-RevId: 8c001b73ff
4 years ago
Megvii Engine Team
b3889938dc
feat(mge/examples): add trace & dump example of cifar10 quantization
GitOrigin-RevId: cfc5e3483a
4 years ago
Megvii Engine Team
2a300415bc
fix(mge): remove a duplicated __init__ call in jit.trace
GitOrigin-RevId: d898f42fc4
4 years ago
Megvii Engine Team
1fe8a21299
fix(mge): fix sublinear memory in jit.trace
GitOrigin-RevId: 190a330a8c
4 years ago
Megvii Engine Team
2df1ab9633
refactor(mge/jit): skip seed when checking equal rng op
GitOrigin-RevId: dae2086b36
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
1569cab0bc
feat(mge/imperative): add shufflenet example
GitOrigin-RevId: d29d8c2bd1
4 years ago
Megvii Engine Team
403a1e7b24
feat(mge/imperative): add graph profiler
GitOrigin-RevId: c73563f337
4 years ago
Megvii Engine Team
a3b2232ba7
feat(mge): add trace.dump
GitOrigin-RevId: ea4c9d33c8
4 years ago
Megvii Engine Team
0e82b959a1
feat(mge/imperative): add sublinear options
GitOrigin-RevId: f0e917f716
4 years ago
Megvii Engine Team
e027dcbf2c
chore(mge): improve symbolic tracing value/shape inference
GitOrigin-RevId: d1a6baac74
4 years ago
Megvii Engine Team
ac3408bfaa
chore(mge): add VarNode.value
GitOrigin-RevId: 1dc0d0c711
4 years ago
Megvii Engine Team
0537cb7471
chore(mge/imperative): fix BackwardGraph for jit.trace
GitOrigin-RevId: 16e86a21d7
4 years ago
Megvii Engine Team
d4bad71190
feat(mge): add jit.trace
GitOrigin-RevId: ec647324c0
4 years ago