Megvii Engine Team
|
d9a46ea47b
|
fix(dnn): correct behaviour of floor div for int tensor
GitOrigin-RevId: 1444f69cce
|
3 years ago |
Megvii Engine Team
|
8918abb17b
|
Revert "test(fastrun): suppress persistent cache test"
This reverts commit 5d6e722deb .
GitOrigin-RevId: 74e184d5c6
|
3 years ago |
Megvii Engine Team
|
cf1db2616e
|
fix(fastrun): replace py_redis with cpp_redis to avoid deadlock
GitOrigin-RevId: 9af7fa5c97
|
3 years ago |
Megvii Engine Team
|
f75261d76f
|
fix(mge/utils): fix module_status error
GitOrigin-RevId: 9e004d98a1
|
3 years ago |
Megvii Engine Team
|
f67aa6466e
|
test(fastrun): suppress persistent cache test
GitOrigin-RevId: 5d6e722deb
|
3 years ago |
Megvii Engine Team
|
2881934cb8
|
feat(dnn/check_non_finite): addmul scale to check_non_finite opr
GitOrigin-RevId: c35a219e52
|
3 years ago |
Megvii Engine Team
|
1f0cc891b0
|
feat(dnn): enable eye to support bool
GitOrigin-RevId: 76d874d5b7
|
3 years ago |
Megvii Engine Team
|
11d75fecb5
|
feat(dnn/check_non_finite): add batch check_non_finite
GitOrigin-RevId: e108133282
|
3 years ago |
Megvii Engine Team
|
7a023c059a
|
feat(mge/traced_module): add optimization api
GitOrigin-RevId: eaa7402640
|
3 years ago |
Megvii Engine Team
|
6c692b26a7
|
feat(mge/traced_module): add some fuse passes
GitOrigin-RevId: 065f9df32e
|
3 years ago |
Megvii Engine Team
|
0185c1a9b9
|
feat(mge/traced_module): add argspec for top TracedModule
GitOrigin-RevId: 8e31a00c7e
|
3 years ago |
Megvii Engine Team
|
e6c271ae46
|
fix(mge/traced_module): fix some bugs for graph surgery
GitOrigin-RevId: 6328a84cbc
|
3 years ago |
Megvii Engine Team
|
7b19bc76fb
|
feat(traced_module): support traced module backward compatible serialization
GitOrigin-RevId: aaa9e51c74
|
3 years ago |
Megvii Engine Team
|
89186edc5d
|
fix(dnn): correct reduce/argmxx/fakequant calculation with nan
GitOrigin-RevId: 7e78bdae91
|
3 years ago |
Megvii Engine Team
|
68cdabd288
|
feat(opr): indexing_multi_axis_vec support nd index
GitOrigin-RevId: 07b1248bdc
|
3 years ago |
Megvii Engine Team
|
5c94db16f2
|
feat(mge/functional): add groups support for conv_transpose2d & 3d
GitOrigin-RevId: b75b792fb4
|
3 years ago |
Megvii Engine Team
|
7591718d20
|
feat(mge): add functional test
GitOrigin-RevId: aa0be62686
|
3 years ago |
Megvii Engine Team
|
846cc13ca8
|
fix(mge/functional): fix default dtype of F.full
GitOrigin-RevId: 5cd34af9bf
|
3 years ago |
Megvii Engine Team
|
0ddabb06cc
|
fix(imperative/astype): fix astype when target dtype is int4
GitOrigin-RevId: 1c5195176d
|
3 years ago |
Megvii Engine Team
|
89ed7ab2ff
|
test(imperative): speed up dtr test
GitOrigin-RevId: 57f092d729
|
3 years ago |
Megvii Engine Team
|
d909950f64
|
refactor(imperative): remove swap in dtr
GitOrigin-RevId: 5c9e42f74a
|
3 years ago |
Megvii Engine Team
|
829f0907e7
|
fix(mge/traced_module): fix insert qat module
GitOrigin-RevId: 35849bc1a2
|
3 years ago |
Megvii Engine Team
|
ba8bd01023
|
fix(mge/traced_module): fix insert module
GitOrigin-RevId: 755e1c68f6
|
3 years ago |
Megvii Engine Team
|
dbca32702a
|
fix(mge/traced_module): fix module dict
GitOrigin-RevId: d7baf00e3c
|
3 years ago |
Megvii Engine Team
|
01d2473ce7
|
fix(mge/traced_module): fix TracedModule flatten
GitOrigin-RevId: 7b15fe492b
|
3 years ago |
Megvii Engine Team
|
23c1fda7e6
|
perf(arm_common): optimize sigmoid
GitOrigin-RevId: 7cb248a15b
|
3 years ago |
Megvii Engine Team
|
a404c508e9
|
feat(mge): support dump with specific format
GitOrigin-RevId: 57a7c0de02
|
3 years ago |
Megvii Engine Team
|
a6fe7f7ff2
|
refactor(mge/traced_module): refactor Node naming rule and merge GetAttr
GitOrigin-RevId: a43ad1273c
|
3 years ago |
Megvii Engine Team
|
ac86d64474
|
fix(mge/module): fix prelu error when use_symbolic_shape is true
GitOrigin-RevId: 25b9c4d41d
|
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
|
2c5a007a92
|
fix(mge/optimizer): allow lr to be 0
GitOrigin-RevId: dabd1fcc33
|
3 years ago |
Megvii Engine Team
|
f7d05db7d6
|
fix(mge/module): fix named_tensors
GitOrigin-RevId: bb5aa1f41d
|
3 years ago |
Megvii Engine Team
|
177dec94c5
|
feat(mgb/opr): add bgr2gray mode for cvtcolor opr
GitOrigin-RevId: d50415b236
|
3 years ago |
Megvii Engine Team
|
cebda6ff8d
|
feat(mge/imperative): add ctc loss
GitOrigin-RevId: e29854a98e
|
3 years ago |
Megvii Engine Team
|
f5cb21ed3a
|
fix(mgb/opr): add non finite check
GitOrigin-RevId: a9fcd0a350
|
3 years ago |
Megvii Engine Team
|
b6142bee9a
|
feat(imperative): support tensor with uint16 date type
GitOrigin-RevId: 57ba0633c7
|
3 years ago |
Megvii Engine Team
|
4e95c13617
|
feat(sgd): sgd supports nesterov momentum
GitOrigin-RevId: 13eda179da
|
3 years ago |
Megvii Engine Team
|
dc28a43409
|
fix(mgb/bn): fix empty tensor input problem and other minor problems
GitOrigin-RevId: 1708b76cb8
|
3 years ago |
Megvii Engine Team
|
c0ccd0ea7e
|
feat(mge/bn): add NHWC support for bn
GitOrigin-RevId: 0a5bb6f72d
|
3 years ago |
chenjiahui
|
6b863cc53a
|
feat(imperative): add pixel_shuffle opr
Closes #211
GitOrigin-RevId: 1846714791
|
3 years ago |
Megvii Engine Team
|
b982be567a
|
feat(mge/imperative): add permutation support for the tensor
GitOrigin-RevId: 7ed0447bfe
|
3 years ago |
Megvii Engine Team
|
3977b7aa0b
|
feat(mgb/shuffle): add shuffle opr
GitOrigin-RevId: 80490a6f84
|
3 years ago |
Megvii Engine Team
|
884a07ffd2
|
fix(test/random): set a random seed for random unit test
GitOrigin-RevId: ad4b01eac7
|
3 years ago |
Megvii Engine Team
|
97b1b7774d
|
feat(mgb): add megbrain layer norm opr with subgraph
GitOrigin-RevId: 9b7fa821f8
|
3 years ago |
Megvii Engine Team
|
f642b05e17
|
test(mge): update traced_module unit test
GitOrigin-RevId: 3948d50d79
|
3 years ago |
Megvii Engine Team
|
43fb2c4afb
|
feat(opr): let roll support empty IO
GitOrigin-RevId: b9a59b623a
|
3 years ago |
Megvii Engine Team
|
b2827cb1b1
|
feat(opr): let Dot, MatrixMul and BatchedMatrixMul support empty input
GitOrigin-RevId: 10a3c5b106
|
3 years ago |
Megvii Engine Team
|
50f738772b
|
feat(opr): let Copy and Identity support empty IO
GitOrigin-RevId: 4e49d8eae8
|
3 years ago |
Megvii Engine Team
|
74cbc10d82
|
feat(opr): let batchnorm support empty IO
GitOrigin-RevId: 219411c80c
|
3 years ago |
Megvii Engine Team
|
8ae61752a0
|
feat(opr): let nms support empty IO
GitOrigin-RevId: 4c51b1aedb
|
3 years ago |