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
Megvii Engine Team
1a1748daf1
feat(opr): let Argsort support empty IO
GitOrigin-RevId: 05fcac6e47
3 years ago
Megvii Engine Team
7234efe1e7
feat(opr): let random op support empty output
GitOrigin-RevId: a2174975aa
3 years ago
Megvii Engine Team
dff7719e87
feat(mge/distributed): add preload host data with op
fix(mge/distributed): change api name with preload
fix(mge/distributed): fix recursive model in preload tensor
fix(mge/distributed): fix recursive when cache contain None
GitOrigin-RevId: 80e2a6dd70
3 years ago