Megvii Engine Team
|
2775f4580c
|
feat(subgraph): subgraph builder supports jit and custom grad
GitOrigin-RevId: e1a1ebdf1c
|
3 years ago |
Megvii Engine Team
|
0bdd0b1467
|
refactor(dispatch): switch to new dispatch system
GitOrigin-RevId: 32dd49a23a
|
3 years ago |
Megvii Engine Team
|
619d78ed86
|
fix(imperative): check async error when getting value
GitOrigin-RevId: 52b8a29932
|
3 years ago |
Megvii Engine Team
|
2afa0af926
|
docs(imperative/jit): add docstring of dump
GitOrigin-RevId: 49eacc0e58
|
3 years ago |
Megvii Engine Team
|
a404c508e9
|
feat(mge): support dump with specific format
GitOrigin-RevId: 57a7c0de02
|
3 years ago |
Megvii Engine Team
|
4f6c5d8f01
|
feat(mge/dump): enable jit.dump to dump with testcase
GitOrigin-RevId: 5dce356452
|
3 years ago |
Megvii Engine Team
|
65d554ed8e
|
refactor(dtr): improve dtr in computing graph
GitOrigin-RevId: d599a08e52
|
3 years ago |
Megvii Engine Team
|
27f2ecc436
|
feat(imperative/opr): add extern c opr and support dump
GitOrigin-RevId: b29e374c60
|
3 years ago |
Megvii Engine Team
|
ee790cb62c
|
fix(imperative/python): fix warp_perspective doc and arange dtype
GitOrigin-RevId: dc4e021ec0
|
3 years ago |
Megvii Engine Team
|
c5f8b58385
|
docs(docstring): transfer to google style
GitOrigin-RevId: a71245c553
|
3 years ago |
Megvii Engine Team
|
8c47c1f149
|
perf(syncbn): reimplement with subgraph
GitOrigin-RevId: 13e7e3d3c0
|
3 years ago |
Megvii Engine Team
|
b3d2a632f3
|
fix(sublinear): fix lb_memory in sublinear
GitOrigin-RevId: fd51023c01
|
3 years ago |
Megvii Engine Team
|
d16c5caf6b
|
fix(mge/dump): fix dump device error with const
GitOrigin-RevId: 9dd8321fd7
|
3 years ago |
Megvii Engine Team
|
1bf18252c4
|
feat(mge/amp): add mix precision autocast support
GitOrigin-RevId: 6fbffc4845
|
4 years ago |
Megvii Engine Team
|
2d42455fa8
|
fix(mge/utils): fix toposort to get definition order
GitOrigin-RevId: 47a26dd6dd
|
4 years ago |
Megvii Engine Team
|
a95f6d4f75
|
perf(trace): add fastpath for const value assert
GitOrigin-RevId: 9a966f257f
|
3 years ago |
Megvii Engine Team
|
9d47c3babd
|
feat(profiler): imperative profiler support tracing
GitOrigin-RevId: b247472feb
|
4 years ago |
Megvii Engine Team
|
54a4d70eb5
|
feat(src/serialization): add support of serializing metadata
GitOrigin-RevId: b563c94451
|
4 years ago |
Megvii Engine Team
|
3e4e4c4604
|
feat(mgb/jit): add graph_opt_config and jit_config interfaces
GitOrigin-RevId: 170d9eeab2
|
4 years ago |
Megvii Engine Team
|
355153e158
|
feat(mge/dtr): add DTR in computing graph
GitOrigin-RevId: 8941810319
|
4 years ago |
Megvii Engine Team
|
241b35a697
|
refactor(ops): remove BackwardGraph op
GitOrigin-RevId: eda20e5760
|
4 years ago |
Megvii Engine Team
|
efc7f04fd3
|
refactor(mge/jit): using static global_enable for apply ctx insted of global variable
GitOrigin-RevId: dd82b53faf
|
4 years ago |
Megvii Engine Team
|
750bde9513
|
refactor(mge/jit): remove is_compiled flag in cpp tensor
GitOrigin-RevId: 15f90af735
|
4 years ago |
Megvii Engine Team
|
4f240ec2d3
|
refactor(mge/jit): make trace return any kind of output
GitOrigin-RevId: fd1265c661
|
4 years ago |
Megvii Engine Team
|
579a3d7786
|
fix(mge/jit): error out if dump bn in training mode
GitOrigin-RevId: edc7ea2962
|
4 years ago |
Megvii Engine Team
|
87d6ff228d
|
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 |