Megvii Engine Team
|
4aaae9950b
|
feat(mge/functional): add python wrapper to resize opr
GitOrigin-RevId: b7cc6dd829
|
4 years ago |
Megvii Engine Team
|
5100317660
|
feat(mge): make Function available at autodiff.Function
GitOrigin-RevId: 7c9ab65351
|
4 years ago |
Megvii Engine Team
|
0a3ca25337
|
fix(mge): fix backward graph optimization
GitOrigin-RevId: 28fd00ac54
|
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
|
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
|
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
|
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
|
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
|
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
|
147cef52a0
|
refactor(mge/imperative): implement new tensor system
GitOrigin-RevId: 2dd4e460ac
|
4 years ago |
Megvii Engine Team
|
9d928e7f83
|
refactor(mge/distributed): sync interpreter for distribtued launcher
GitOrigin-RevId: 8a88c272a1
|
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
|
ee4ea7fdc8
|
test(distributed/test): make distributed test more stronger
GitOrigin-RevId: 085fd1dcfd
|
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
|
638ab52fdc
|
feat(mge/imperative): simulates scalar
GitOrigin-RevId: e81630e256
|
4 years ago |
Megvii Engine Team
|
176268d20c
|
test(mge): disable gc in leak check
GitOrigin-RevId: b650a46e02
|
4 years ago |
Megvii Engine Team
|
50c4daac85
|
feat(mge/interpreter): add async_level mechanism for Interpreter
GitOrigin-RevId: 8615a23b75
|
4 years ago |
Megvii Engine Team
|
be511a56f9
|
fix(mge/imperative): fix tensor astype failed for quantized type
GitOrigin-RevId: 383458acbf
|
4 years ago |
Megvii Engine Team
|
fccb2510c4
|
perf(mge): add opdef for elemwise and batchnorm
GitOrigin-RevId: d51fad9867
|
4 years ago |
Megvii Engine Team
|
9748aebeea
|
refactor(mge): tensor_shape -> symbolic_shape
GitOrigin-RevId: 366dc048bf
|
4 years ago |
Megvii Engine Team
|
0d4568d6f8
|
feat(mge): rename broadcast -> broadcast_to
GitOrigin-RevId: 82f46ad2c2
|
4 years ago |
Megvii Engine Team
|
1e2117f66d
|
feat(mge): remove add_axis and remove_axis
GitOrigin-RevId: 59611d43f9
|
4 years ago |
Megvii Engine Team
|
912d733ea9
|
fix(dnn): support bool for IndexingMultiAxisVec
GitOrigin-RevId: ddcfaa06b0
|
4 years ago |
Megvii Engine Team
|
8109cc4bd1
|
fix(imperative): set exception which worker thread throws for rendezvous
GitOrigin-RevId: f583888fdf
|
4 years ago |
Megvii Engine Team
|
6ce796aec4
|
test(mge): replace `assertTensorClose` with `np.testing.assert_allclose`
GitOrigin-RevId: 2bf8c42cfd
|
4 years ago |
Megvii Engine Team
|
f63bca032c
|
test(mge/dtype): add quant and lowbit dtype test
GitOrigin-RevId: 97ca1a393a
|
4 years ago |
Megvii Engine Team
|
1965276845
|
fix(mge): fix bool.sum()
GitOrigin-RevId: 62c482db40
|
4 years ago |
Megvii Engine Team
|
81b6a73382
|
fix(mge): fix bug of tensor.T
GitOrigin-RevId: 9fe9347b00
|
4 years ago |
Megvii Engine Team
|
9005cf74d2
|
fix(mge/imperative): fix tracer leak
GitOrigin-RevId: 1a8ac20b46
|
4 years ago |
Megvii Engine Team
|
a8c75ee530
|
refactor(mge/functional): make docs better
GitOrigin-RevId: acb287d48b
|
4 years ago |