Megvii Engine Team
|
4aaae9950b
|
feat(mge/functional): add python wrapper to resize opr
GitOrigin-RevId: b7cc6dd829
|
4 years ago |
Megvii Engine Team
|
df976782fa
|
feat(mge/functional): add matinv
GitOrigin-RevId: d4fa8a8277
|
4 years ago |
Megvii Engine Team
|
5100317660
|
feat(mge): make Function available at autodiff.Function
GitOrigin-RevId: 7c9ab65351
|
4 years ago |
Megvii Engine Team
|
abe3c165ba
|
feat(mge): tensorrt runtime opr
GitOrigin-RevId: 2fdd00adcb
|
4 years ago |
Megvii Engine Team
|
c418d3cd95
|
fix(mge/data): add timeout event
GitOrigin-RevId: 43f2ba1456
|
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
|
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
|
0b4918b25e
|
test(mge/quantization): classmethod `from_float_module` of qat module
GitOrigin-RevId: 95c3d45f83
|
4 years ago |
Megvii Engine Team
|
5c7d48cdb9
|
fix(mge/functional): fix tensor split
GitOrigin-RevId: 0a112ab0bd
|
4 years ago |
Megvii Engine Team
|
d3bfb0e983
|
fix(mge): fix trace exit code and reformat
GitOrigin-RevId: 145c06b7e7
|
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
|
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
|
a85531dd0f
|
feat(mgb/opr): add tqt opr
GitOrigin-RevId: 49c62cd532
|
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
|
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
|
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
|
b9762d714c
|
fix(mge): make parampack run with tensor symbolic shape
GitOrigin-RevId: 6fc313785d
|
4 years ago |
Megvii Engine Team
|
4d75f691a0
|
feat(mge): restore remote send/recv
GitOrigin-RevId: 8b78fd5591
|
4 years ago |
Megvii Engine Team
|
556e022215
|
refactor(mge/trace): remove apply on varnode
GitOrigin-RevId: 0185244c1f
|
4 years ago |
Megvii Engine Team
|
645fc6f0c8
|
test(mge/autodiff): add test for weird error
GitOrigin-RevId: e95109b845
|
4 years ago |
Megvii Engine Team
|
60c7d62a4c
|
refactor(imperative): remove multidispatch, raw_tensor, register
GitOrigin-RevId: ca5a6ed8eb
|
4 years ago |
Megvii Engine Team
|
b5e46ae92f
|
feat(mge): restore Function
GitOrigin-RevId: dd455238ba
|
4 years ago |
Megvii Engine Team
|
61f65cd495
|
test(mge): fix megbrain_graph/cgtools test
GitOrigin-RevId: 33ac776b56
|
4 years ago |
Megvii Engine Team
|
87f4b46e3f
|
perf(mge/imperative): move convert_inputs from python to C++
GitOrigin-RevId: baef3d348c
|
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
|
15e8e7df76
|
refactor(mge/imperative): migrate to new core implementation
* swap/drop
* config/set_async_level
* _dev_tensor
* sync
GitOrigin-RevId: 850fb98852
|
4 years ago |
Megvii Engine Team
|
34c705fcbf
|
refactor(mge/imperative): move detach into C++
GitOrigin-RevId: 8c0d86cbbf
|
4 years ago |
Megvii Engine Team
|
147cef52a0
|
refactor(mge/imperative): implement new tensor system
GitOrigin-RevId: 2dd4e460ac
|
4 years ago |
Megvii Engine Team
|
b5ec83c53f
|
test(mge/autodiff): validate grad is set to None when no dependency
GitOrigin-RevId: 01ad51a222
|
4 years ago |
Megvii Engine Team
|
9d928e7f83
|
refactor(mge/distributed): sync interpreter for distribtued launcher
GitOrigin-RevId: 8a88c272a1
|
4 years ago |
Megvii Engine Team
|
94796060b6
|
test(mge/quantization): qat module, quantized op and module
GitOrigin-RevId: 17b28060cc
|
4 years ago |
Megvii Engine Team
|
ab9f44f15c
|
feat(mge/quantization): add support for easyquant
GitOrigin-RevId: 060d908349
|
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
|
57546b4c3d
|
test(mge/distributed): fix test skip condition error
GitOrigin-RevId: f540c53302
|
4 years ago |
Megvii Engine Team
|
ee4ea7fdc8
|
test(distributed/test): make distributed test more stronger
GitOrigin-RevId: 085fd1dcfd
|
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
|
6c4841e807
|
fix(mge/quantization): `disable_fake_quant` does not work correctly
GitOrigin-RevId: 0c568d3335
|
4 years ago |
Megvii Engine Team
|
5a01de7851
|
fix(mge): fix scalar transpose
GitOrigin-RevId: c2b9e025c7
|
4 years ago |
Megvii Engine Team
|
638ab52fdc
|
feat(mge/imperative): simulates scalar
GitOrigin-RevId: e81630e256
|
4 years ago |