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