Megvii Engine Team
|
c418d3cd95
|
fix(mge/data): add timeout event
GitOrigin-RevId: 43f2ba1456
|
4 years ago |
Megvii Engine Team
|
c53abcdf1d
|
chore(mge): minor improvements related to grad
GitOrigin-RevId: 102467d79d
|
4 years ago |
Megvii Engine Team
|
0a3ca25337
|
fix(mge): fix backward graph optimization
GitOrigin-RevId: 28fd00ac54
|
4 years ago |
Megvii Engine Team
|
ea8eb4cf72
|
feat(mge/distributed): scalar support for distributed functions
GitOrigin-RevId: 53f3575baf
|
4 years ago |
Megvii Engine Team
|
b83c77e105
|
test(mge/test_save_load): remove test model file
GitOrigin-RevId: 893b82b1c9
|
4 years ago |
Megvii Engine Team
|
c522991385
|
test(mge/core): add host compute test, fix async_level and swap/drop test
GitOrigin-RevId: f9b5a83507
|
4 years ago |
Megvii Engine Team
|
c8e04ce4df
|
fix(imperative/interpreter): use info->h_value for host compute
Rather than use host_only Tensor and modify Put's behavior.
GitOrigin-RevId: f890d66acb
|
4 years ago |
Megvii Engine Team
|
a5a606792e
|
feat(imperative/interpreter): add more dispatch mode in apply_op
GitOrigin-RevId: 2663504470
|
4 years ago |
Megvii Engine Team
|
248d8bf0dc
|
feat(imperative/ops): improve infer attrs validate function
GitOrigin-RevId: 6fab3b1402
|
4 years ago |
Megvii Engine Team
|
8ed2077bcb
|
fix(imperative/interpreter): reset m_waitee on exception
GitOrigin-RevId: ecf5245d9c
|
4 years ago |
Megvii Engine Team
|
8fe865d88a
|
feat(imperative/ops): add infer_output_attrs for Reshape
GitOrigin-RevId: 9150d7f84d
|
4 years ago |
Megvii Engine Team
|
267d612723
|
fix(imperative/mini_graph): fix no_err value in shape infer
GitOrigin-RevId: 087d15db66
|
4 years ago |
Megvii Engine Team
|
bc3561c1ee
|
fix(imperative/ops): fix inplace_add's output desc, remove value
GitOrigin-RevId: b294eead0a
|
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
|
55042195d4
|
chore(winograd): add Convolutionv2 param
GitOrigin-RevId: 1a9e2ea340
|
4 years ago |
Megvii Engine Team
|
7191c4bd9f
|
fix(mge/data): support timeout for serial stream dataloader
GitOrigin-RevId: 1ae5a8cfda
|
4 years ago |
Megvii Engine Team
|
b7ed0cb850
|
fix(mge/quantization): replace `_reset` with "=" in observer
GitOrigin-RevId: ed6af9b98d
|
4 years ago |
Megvii Engine Team
|
5d0c9bef4c
|
chore(mge/version): change dev version to 1.3
GitOrigin-RevId: b5e5327a46
|
4 years ago |
Megvii Engine Team
|
0b4918b25e
|
test(mge/quantization): classmethod `from_float_module` of qat module
GitOrigin-RevId: 95c3d45f83
|
4 years ago |
Megvii Engine Team
|
e9a7be46b1
|
feat(mgb): adapt to imperative runtime
GitOrigin-RevId: 3bccc17b62
|
4 years ago |
Megvii Engine Team
|
08cc10324e
|
fix(imperative): fix persistent_cache
GitOrigin-RevId: 8f7bb5899f
|
4 years ago |
Megvii Engine Team
|
998f71a8c9
|
perf(mge/imperative): do not busy wait in imperative's queue
GitOrigin-RevId: de8db51096
|
4 years ago |
Megvii Engine Team
|
278b2baa8c
|
perf(mge): add memory optimization for backward graph
precompute ops in forward to reduce saved tensor size
GitOrigin-RevId: d67043ba82
|
4 years ago |
Megvii Engine Team
|
5c7d48cdb9
|
fix(mge/functional): fix tensor split
GitOrigin-RevId: 0a112ab0bd
|
4 years ago |
Megvii Engine Team
|
82847bc70a
|
docs(mge): add conv2d output shape computation formula
GitOrigin-RevId: 7ee476a80d
|
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
|
a90c937d22
|
feat(interpreter): add command buffer for inplace
GitOrigin-RevId: 020d1e88d4
|
4 years ago |
Megvii Engine Team
|
09af925fb6
|
fix(mge): fix cpp trace function release
GitOrigin-RevId: 924f945c21
|
4 years ago |
Megvii Engine Team
|
6919127fd4
|
fix(mge): fix event pool destructor order issue and do full sync at python exit
GitOrigin-RevId: b9fa1f45b1
|
4 years ago |
Megvii Engine Team
|
60459bd79b
|
fix(mge/pytest): remove sync in pytest teardown & add full_sync
GitOrigin-RevId: ee278a8ac1
|
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
|
89b6dbc74c
|
fix(mge/imperative): import subpackages
GitOrigin-RevId: 342f6c87cc
|
4 years ago |
Megvii Engine Team
|
a85531dd0f
|
feat(mgb/opr): add tqt opr
GitOrigin-RevId: 49c62cd532
|
4 years ago |
Megvii Engine Team
|
44742e32e9
|
fix(mge/api): check input dim of dot and mark output as scalar
GitOrigin-RevId: a3ba7e099c
|
4 years ago |
Megvii Engine Team
|
697f70c011
|
feat(mge/pytest): add more tests for specialized grad rules
GitOrigin-RevId: 509ef5a220
|
4 years ago |
Megvii Engine Team
|
cf3f58cb9b
|
fix(mge/autodiff): fix segfault when grad is nullptr
GitOrigin-RevId: 6139212bfd
|
4 years ago |
Megvii Engine Team
|
288c2e08b5
|
fix(mge/autodiff): fix expand_dims and grad rule fallback
GitOrigin-RevId: 4aae771222
|
4 years ago |
Megvii Engine Team
|
05c739b846
|
refactor(mge/data): rename `MapDataset` to `Dataset`
GitOrigin-RevId: 6262561355
|
4 years ago |
Megvii Engine Team
|
a892e5d0e4
|
perf(mge): add more specialized grad rules
GitOrigin-RevId: f88809a6d7
|
4 years ago |
Megvii Engine Team
|
e9e5f442a7
|
fix(mge): expand custom op before trace
GitOrigin-RevId: 725a5b87cb
|
4 years ago |
Megvii Engine Team
|
3faba54f28
|
fix(mge): fix segfault with Function returning unused grads
GitOrigin-RevId: 0cce845923
|
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
|
aba0acc797
|
fix(sdk): add AssertEqual opr, fix dump_with_testcase_mge
GitOrigin-RevId: 6f797570b6
|
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
|
b9762d714c
|
fix(mge): make parampack run with tensor symbolic shape
GitOrigin-RevId: 6fc313785d
|
4 years ago |