Megvii Engine Team
|
2c2beac915
|
fix(mge/utils): fix bug of VarNode inplace operations
GitOrigin-RevId: fa9eec7079
|
4 years ago |
Megvii Engine Team
|
4fa9096d20
|
feat(megbrain): add correlation opr
GitOrigin-RevId: 6d44598891
|
4 years ago |
Megvii Engine Team
|
67167cb39d
|
feat(imperative/autodiff): add grad clip
GitOrigin-RevId: f344f4a233
|
4 years ago |
Megvii Engine Team
|
7d18102795
|
fix(imperative/tensor): init m_offset when constructing a Tensor with DeviceTensorND
GitOrigin-RevId: b340e27c47
|
4 years ago |
Megvii Engine Team
|
45b9b5ed14
|
test(mge/distributed): add gather scatter reduce broadcast grad test
GitOrigin-RevId: 8245e11f1d
|
4 years ago |
Megvii Engine Team
|
27ff38dd21
|
feat(functional/loss): add reduction choices to loss functions
GitOrigin-RevId: a29e6bb4cf
|
4 years ago |
Megvii Engine Team
|
0a1bd90f6b
|
test(param_pack): more cases for param pack concat
GitOrigin-RevId: 0700b548ab
|
4 years ago |
Megvii Engine Team
|
acb239927b
|
fix(mge/jit): error out if dump bn in training mode
GitOrigin-RevId: edc7ea2962
|
4 years ago |
Megvii Engine Team
|
a405461c01
|
fix(mge/elemwise): fix problem that elemwise.mode is not comparable with string mode
GitOrigin-RevId: 82e39be0a9
|
4 years ago |
Megvii Engine Team
|
b6dc4c824d
|
fix(tensor): del valid tensors when compnode finalizing
GitOrigin-RevId: bace1f2b51
|
4 years ago |
Megvii Engine Team
|
2c61b8bff4
|
feat(mge/functional): argmin and argmax support negtive axis
GitOrigin-RevId: a1bd1102a6
|
4 years ago |
Megvii Engine Team
|
cb84efa74f
|
fix(mge): fix dumping backward graph
GitOrigin-RevId: 430f110053
|
4 years ago |
Megvii Engine Team
|
bb94f2aa3d
|
feat(mge/module): add __repr__ method for qat and quantized module
GitOrigin-RevId: 0d78cbab9a
|
4 years ago |
Megvii Engine Team
|
5dbf3612b2
|
test(mge/optimizer): update optimizer test to make sure grad not change
GitOrigin-RevId: e207672116
|
4 years ago |
Megvii Engine Team
|
a3e098c80c
|
fix(mge/utils): fix network multiple outputs issue
GitOrigin-RevId: d22e639cd3
|
4 years ago |
Megvii Engine Team
|
0d332cf005
|
feat(imperative/opr): rebase rng refactoring to dev & add python module
GitOrigin-RevId: ee5984c52d
|
4 years ago |
Megvii Engine Team
|
596dc982bb
|
feat(imperative): add rng opdef
GitOrigin-RevId: b62dcea7f5
|
4 years ago |
Megvii Engine Team
|
800f119c2e
|
feat(mge/tools): module_status-add-functions
BREAKING CHANGE:
GitOrigin-RevId: ced3da3a12
|
4 years ago |
Megvii Engine Team
|
11de49b943
|
feat(mge/module): python wrapper for conv_transpose3d
GitOrigin-RevId: 61097b8713
|
4 years ago |
Megvii Engine Team
|
331995e7f6
|
refactor(imperative): alloc enum type class on heap
GitOrigin-RevId: d2b2acea22
|
4 years ago |
Megvii Engine Team
|
45a031fec3
|
test(dist): remove redundant assert for reduce_sum and gather
GitOrigin-RevId: bc7393bb63
|
4 years ago |
Megvii Engine Team
|
d4eb389042
|
feat(distributed): auto detect device and backend when init group
GitOrigin-RevId: 90be2d5b4d
|
4 years ago |
Megvii Engine Team
|
9be7cb8924
|
test(mge/module): add module reset attribute test
GitOrigin-RevId: 6c9adc4a70
|
4 years ago |
Megvii Engine Team
|
976b351ada
|
feat(functional/nn): support F.warp_perspective with `mat_idx`
GitOrigin-RevId: 66910c8bd8
|
4 years ago |
Megvii Engine Team
|
fe99cdc794
|
feat(interpreter): add dynamic sublinear
GitOrigin-RevId: 7de54fe743
|
4 years ago |
Megvii Engine Team
|
dcff115ef8
|
fix(distributed/launcher): fetch data early to avoid dead lock in launcher
GitOrigin-RevId: 9abcc956ef
|
4 years ago |
Megvii Engine Team
|
7e22e9f06e
|
feat(optimzer): add AdamW
GitOrigin-RevId: e608b5d5b9
|
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
|
4f0e6eae35
|
fix(mge/trace): re-open gopt level in trace
GitOrigin-RevId: 5ebc712690
|
4 years ago |
Megvii Engine Team
|
e474994f46
|
feat(imperative/jit): catch input tensors name when tracing
GitOrigin-RevId: 9c69254866
|
4 years ago |
Megvii Engine Team
|
aed681d325
|
feat(imperative/utils): optimize the naming rules
GitOrigin-RevId: 329bac640a
|
4 years ago |
Megvii Engine Team
|
04b1a45af4
|
fix(dnn): fix cudnn crash when finalize called after cudnn dtor
GitOrigin-RevId: b0ad639921
|
4 years ago |
Megvii Engine Team
|
0ad85a41ac
|
test(mge): do not ignore data test
GitOrigin-RevId: 937d24df5e
|
4 years ago |
Megvii Engine Team
|
31e4bf2c34
|
fix(mge/optimizer): fix optimizer's state_dict bug
GitOrigin-RevId: 67fb112fb8
|
4 years ago |
Megvii Engine Team
|
dc96f6aa49
|
fix(mge/quantization): fix quantized concat forward problem
GitOrigin-RevId: dc21b340d1
|
4 years ago |
Megvii Engine Team
|
48a7d333c3
|
fix(mge/module): fix auto-naming error when there are containers in the module
GitOrigin-RevId: 6b8f043321
|
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
|
efe264b751
|
test(mge/utils): skip network node convbias test in gpu
GitOrigin-RevId: 9d06cf6c5a
|
4 years ago |
Megvii Engine Team
|
cdbb4a20ff
|
fix(mge/tensor): fix tensor's serialization behavior
GitOrigin-RevId: 4d74a4b46e
|
4 years ago |
Megvii Engine Team
|
a3ea1f153c
|
feat(mgb/opr): add fast profile and combined Execution strategy
GitOrigin-RevId: 843dc3a790
|
4 years ago |
Megvii Engine Team
|
9c90ce8c33
|
fix(mge/quantization): fix `quantize` and `quantize_qat`'s `set_expand_structure` arguments
GitOrigin-RevId: c61633095d
|
4 years ago |
Megvii Engine Team
|
e6b06914f7
|
feat(mge/tensor): support non-Tensor value in `_reset` and remove depreciated tests
GitOrigin-RevId: faf6c78aa8
|
4 years ago |
Megvii Engine Team
|
5a38ad3920
|
feat(mge/utils): add get/set_expand_structure to deal with complex key
GitOrigin-RevId: 4d1b952068
|
4 years ago |
Megvii Engine Team
|
3f2e15f534
|
fix(mge/logger): fix set_mgb_log_level
GitOrigin-RevId: 6c94c51483
|
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
|
bfb9def988
|
test(mge/test): reorganize utils test files
GitOrigin-RevId: 08d27a8b1d
|
4 years ago |