Megvii Engine Team
|
d2673c5abf
|
fix(ci/windows): add windows cuda test
GitOrigin-RevId: 706be83032
|
4 years ago |
Megvii Engine Team
|
1525a02530
|
feat(mge/module): add python wrapper for unfold
GitOrigin-RevId: 562103186f
|
4 years ago |
Megvii Engine Team
|
13b15fb08c
|
feat(megbrain): add correlation opr
GitOrigin-RevId: 6d44598891
|
4 years ago |
Megvii Engine Team
|
226f24fa8e
|
test(mge/distributed): add gather scatter reduce broadcast grad test
GitOrigin-RevId: 8245e11f1d
|
4 years ago |
Megvii Engine Team
|
b1bf193e4d
|
feat(functional/loss): add reduction choices to loss functions
GitOrigin-RevId: a29e6bb4cf
|
4 years ago |
Megvii Engine Team
|
717b88e684
|
fix(mge/elemwise): fix problem that elemwise.mode is not comparable with string mode
GitOrigin-RevId: 82e39be0a9
|
4 years ago |
Megvii Engine Team
|
984d85ca26
|
feat(mge/functional): argmin and argmax support negtive axis
GitOrigin-RevId: a1bd1102a6
|
4 years ago |
Megvii Engine Team
|
6972896933
|
fix(mge/utils): fix network multiple outputs issue
GitOrigin-RevId: d22e639cd3
|
4 years ago |
Megvii Engine Team
|
4d9073fcba
|
test(dist): remove redundant assert for reduce_sum and gather
GitOrigin-RevId: bc7393bb63
|
4 years ago |
Megvii Engine Team
|
dee5a10af9
|
feat(distributed): auto detect device and backend when init group
GitOrigin-RevId: 90be2d5b4d
|
4 years ago |
Megvii Engine Team
|
933188aa93
|
feat(functional/nn): support F.warp_perspective with `mat_idx`
GitOrigin-RevId: 66910c8bd8
|
4 years ago |
Megvii Engine Team
|
65af9cfcdc
|
refactor(mge): use lower case for default string parameters in functional and module
GitOrigin-RevId: dbc1f27ff7
|
4 years ago |
Megvii Engine Team
|
9dbe71dd1a
|
feat(mge/utils): add array method for varnode
GitOrigin-RevId: 6e4d05b475
|
4 years ago |
Megvii Engine Team
|
988f897ee2
|
test(mge): make test stable
GitOrigin-RevId: 5de478f8b1
|
4 years ago |
Megvii Engine Team
|
fa1ca0ea6c
|
fix(imperative/opr): fix apply_on_var_node for broadcast
GitOrigin-RevId: 686fff4f73
|
4 years ago |
Megvii Engine Team
|
d970b85deb
|
feat(functional): add conv3d
GitOrigin-RevId: c06ac77c6c
|
4 years ago |
Megvii Engine Team
|
46cad4d3c8
|
feat(functional/ops): add _assert_equal
GitOrigin-RevId: b7ce4158b7
|
4 years ago |
Megvii Engine Team
|
dcfb6a537e
|
refactor(mge/functional): move functional api
GitOrigin-RevId: 9cd3e09996
|
4 years ago |
Megvii Engine Team
|
fe9c6e26a8
|
feat(imperative/opr): deprecate resize op and make it as a special case of interpolate
GitOrigin-RevId: a5668c5779
|
4 years ago |
Megvii Engine Team
|
fa4bf16800
|
feat(mge/functional): add repeat and tile opr
GitOrigin-RevId: a20d4b6fb0
|
4 years ago |
Megvii Engine Team
|
0c37a588ca
|
fix(mge/functional): fix F.ones when input is a tensor of scalar type
GitOrigin-RevId: 6d01d6b58d
|
4 years ago |
Megvii Engine Team
|
1d7dd00144
|
feat(mge/quantization): add QParams and QuantDtypeMeta for quantization data structure
GitOrigin-RevId: df3416fe13
|
4 years ago |
Megvii Engine Team
|
d0f70a445d
|
fix(imperative/utils): module parameters' name do not have scope after dumping
GitOrigin-RevId: 1497272294
|
4 years ago |
Megvii Engine Team
|
cbff4d7c1a
|
fix(imperative/ops): fix infer_output_attrs_fallible for reshape
GitOrigin-RevId: a93567d79a
|
4 years ago |
Megvii Engine Team
|
02df634da2
|
fix(mge/quantization): set ``q_dict`` as an instance property
GitOrigin-RevId: 2f32008aad
|
4 years ago |
Megvii Engine Team
|
cd02d7c85e
|
feat(mge/opr): support deformable conv2d/psroi_pooling
GitOrigin-RevId: 501cadda76
|
4 years ago |
Megvii Engine Team
|
4485e780ae
|
fix(mge/functional): fix linspace device and open other trace tests
GitOrigin-RevId: 4667c4adec
|
4 years ago |
Megvii Engine Team
|
334eda8717
|
refactor(mge): test trace inside opr_test
GitOrigin-RevId: 2cf1135c1c
|
4 years ago |
Megvii Engine Team
|
938152af63
|
fix(mge/functional): convert input type to float32 for more elemwise op
GitOrigin-RevId: cf3bf8cb80
|
4 years ago |
Megvii Engine Team
|
194660462c
|
feat(mge/funcitonal): add cvt_color opr python interface
GitOrigin-RevId: 29e069fb23
|
4 years ago |
Megvii Engine Team
|
ff05667b48
|
test(dist): refactor distributed test with fixtures
GitOrigin-RevId: e69acb72da
|
4 years ago |
Megvii Engine Team
|
e3a3e0cd3c
|
feat(mge/functional): add warp_affine
GitOrigin-RevId: 2b333ccd12
|
4 years ago |
Megvii Engine Team
|
364afec033
|
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
|
4 years ago |
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
|
ea8eb4cf72
|
feat(mge/distributed): scalar support for distributed functions
GitOrigin-RevId: 53f3575baf
|
4 years ago |
Megvii Engine Team
|
5c7d48cdb9
|
fix(mge/functional): fix tensor split
GitOrigin-RevId: 0a112ab0bd
|
4 years ago |
Megvii Engine Team
|
4d75f691a0
|
feat(mge): restore remote send/recv
GitOrigin-RevId: 8b78fd5591
|
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
|
9d928e7f83
|
refactor(mge/distributed): sync interpreter for distribtued launcher
GitOrigin-RevId: 8a88c272a1
|
4 years ago |
Megvii Engine Team
|
ee4ea7fdc8
|
test(distributed/test): make distributed test more stronger
GitOrigin-RevId: 085fd1dcfd
|
4 years ago |
Megvii Engine Team
|
442058aead
|
feat(mge/quantize): add batch matmul activation module for inference
GitOrigin-RevId: f6dbd1f4c0
|
4 years ago |
Megvii Engine Team
|
82b0f67770
|
fix(mge/core): fix dtype promotion issue for quantized dtype
GitOrigin-RevId: 9d09c8fa6f
|
4 years ago |
Megvii Engine Team
|
9cb3c07cd4
|
feat(mge/functional): add elemwise mode support string input
GitOrigin-RevId: 57be5cec7b
|
4 years ago |
Megvii Engine Team
|
8fad00a158
|
feat(mge/functional/nn): add conv1d padding
GitOrigin-RevId: 1bbfd36b96
|
4 years ago |
Megvii Engine Team
|
28dbadf729
|
feat(imperative): add remap opr
GitOrigin-RevId: 1fa7e18d4b
|
4 years ago |
Megvii Engine Team
|
1217801133
|
perf(mge): add opdef for broadcast
GitOrigin-RevId: 92f0af29eb
|
4 years ago |
Megvii Engine Team
|
dfc3935d06
|
fix(pytest/windows/impertive): fix windows wrong test skip
GitOrigin-RevId: ca0f04cfe7
|
4 years ago |
Megvii Engine Team
|
215f88f373
|
fix(dnn/argmxx): fix argmxx on inf
GitOrigin-RevId: 740f67b73a
|
4 years ago |