Megvii Engine Team
cb84efa74f
fix(mge): fix dumping backward graph
GitOrigin-RevId: 430f110053
4 years ago
Megvii Engine Team
4f0e6eae35
fix(mge/trace): re-open gopt level in trace
GitOrigin-RevId: 5ebc712690
4 years ago
Megvii Engine Team
e474994f46
feat(imperative/jit): catch input tensors name when tracing
GitOrigin-RevId: 9c69254866
4 years ago
Megvii Engine Team
aed681d325
feat(imperative/utils): optimize the naming rules
GitOrigin-RevId: 329bac640a
4 years ago
Megvii Engine Team
c8697a7005
feat(imperative/src): python wrapper for cambricon and atlas runtime opr
GitOrigin-RevId: bd969d1339
4 years ago
Megvii Engine Team
46cad4d3c8
feat(functional/ops): add _assert_equal
GitOrigin-RevId: b7ce4158b7
4 years ago
Megvii Engine Team
1d7dd00144
feat(mge/quantization): add QParams and QuantDtypeMeta for quantization data structure
GitOrigin-RevId: df3416fe13
4 years ago
Megvii Engine Team
d0f70a445d
fix(imperative/utils): module parameters' name do not have scope after dumping
GitOrigin-RevId: 1497272294
4 years ago
Megvii Engine Team
6fb19b664b
feat(imperative/src): name operators automatically when tracing
GitOrigin-RevId: ff8eb003c5
4 years ago
Megvii Engine Team
f211124865
feat(mge/trace): support dict return value processing in trace
GitOrigin-RevId: 5b1c08848b
4 years ago
Megvii Engine Team
a09fc5f784
fix(mgb/serialization): disable inplace arith graph opt in graph load
GitOrigin-RevId: d63baf8356
4 years ago
Megvii Engine Team
364afec033
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
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
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
c49427d15a
feat(imperative): add inplace add_update option in optimizer
GitOrigin-RevId: b8feb49321
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
556e022215
refactor(mge/trace): remove apply on varnode
GitOrigin-RevId: 0185244c1f
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
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