Megvii Engine Team
|
0245594156
|
test(autograd): test jvp emulated by 2nd grad
GitOrigin-RevId: 47114fcd99
|
4 years ago |
Megvii Engine Team
|
8480302da8
|
fix(autograd): make higher order grad experimental
GitOrigin-RevId: 81e1eb0ebf
|
4 years ago |
Megvii Engine Team
|
72531f2b4a
|
test(autograd): add more tests for higher order grad
GitOrigin-RevId: 5fc308f87a
|
4 years ago |
Megvii Engine Team
|
522e556b83
|
feat(autodiff): support higher order grad
GitOrigin-RevId: 86390d2179
|
4 years ago |
Megvii Engine Team
|
5198b7834c
|
fix(mge/functional): fix expand_dims for scalar
GitOrigin-RevId: 253ea608f7
|
4 years ago |
Megvii Engine Team
|
933dd9a497
|
feat(mge/distributed): add cuda env check before forked thread
style(core/comp_node): reformat code
GitOrigin-RevId: 372452a8eb
|
4 years ago |
Megvii Engine Team
|
a5060a2bfe
|
feat(mgb/opr): add check_has_inf kernel and opr
GitOrigin-RevId: 0d042dbfce
|
4 years ago |
Megvii Engine Team
|
c64b1c94cd
|
feat(imperative/functional): add roll in functional
GitOrigin-RevId: ff630f1fe7
|
4 years ago |
Megvii Engine Team
|
c78a7848a1
|
test(profiler): simple tests for profiler
GitOrigin-RevId: 92073ce357
|
4 years ago |
Megvii Engine Team
|
54a4d70eb5
|
feat(src/serialization): add support of serializing metadata
GitOrigin-RevId: b563c94451
|
4 years ago |
Megvii Engine Team
|
3e4e4c4604
|
feat(mgb/jit): add graph_opt_config and jit_config interfaces
GitOrigin-RevId: 170d9eeab2
|
4 years ago |
Megvii Engine Team
|
34262d904c
|
fix(imperative): fix the size of blob with offset
GitOrigin-RevId: bb1736dfe1
|
4 years ago |
Megvii Engine Team
|
787f187e7d
|
fix(imperative/src): fix dot backward error
GitOrigin-RevId: 02ba44a0e6
|
4 years ago |
Megvii Engine Team
|
e275dfeca1
|
feat(imperative/python): support pooling mode "average" for avg pool2d module
GitOrigin-RevId: 9fe442129f
|
4 years ago |
Megvii Engine Team
|
d2673c5abf
|
fix(ci/windows): add windows cuda test
GitOrigin-RevId: 706be83032
|
4 years ago |
Megvii Engine Team
|
e417798f9e
|
fix(mge): correct pytype when calling apply from python
GitOrigin-RevId: 6abfa06ada
|
4 years ago |
Megvii Engine Team
|
1525a02530
|
feat(mge/module): add python wrapper for unfold
GitOrigin-RevId: 562103186f
|
4 years ago |
Megvii Engine Team
|
46d96478a1
|
feat(mge/tools): optimize statistical tools
BREAKING CHANGE:
GitOrigin-RevId: cd2a1acd11
|
4 years ago |
Megvii Engine Team
|
7f9d441cbb
|
docs(mge/module): add note about weight shape in conv
GitOrigin-RevId: 43e1f15968
|
4 years ago |
Megvii Engine Team
|
8e5bf94857
|
fix(mge/utils): fix bug of VarNode inplace operations
GitOrigin-RevId: fa9eec7079
|
4 years ago |
Megvii Engine Team
|
13b15fb08c
|
feat(megbrain): add correlation opr
GitOrigin-RevId: 6d44598891
|
4 years ago |
Megvii Engine Team
|
259894250b
|
feat(imperative/autodiff): add grad clip
GitOrigin-RevId: f344f4a233
|
4 years ago |
Megvii Engine Team
|
9b69a02fb2
|
fix(imperative/tensor): init m_offset when constructing a Tensor with DeviceTensorND
GitOrigin-RevId: b340e27c47
|
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
|
7af1ec66db
|
fix(tensor): del valid tensors when compnode finalizing
GitOrigin-RevId: bace1f2b51
|
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
|
87d6ff228d
|
fix(mge): fix dumping backward graph
GitOrigin-RevId: 430f110053
|
4 years ago |
Megvii Engine Team
|
f31752d5cc
|
feat(mge/module): add __repr__ method for qat and quantized module
GitOrigin-RevId: 0d78cbab9a
|
4 years ago |
Megvii Engine Team
|
6972896933
|
fix(mge/utils): fix network multiple outputs issue
GitOrigin-RevId: d22e639cd3
|
4 years ago |
Megvii Engine Team
|
13e6ea349d
|
feat(imperative/opr): rebase rng refactoring to dev & add python module
GitOrigin-RevId: ee5984c52d
|
4 years ago |
Megvii Engine Team
|
cded8ef1a6
|
feat(imperative): add rng opdef
GitOrigin-RevId: b62dcea7f5
|
4 years ago |
Megvii Engine Team
|
c3a1ac3dcf
|
feat(mge/tools): module_status-add-functions
BREAKING CHANGE:
GitOrigin-RevId: ced3da3a12
|
4 years ago |
Megvii Engine Team
|
e18afa0b05
|
feat(mge/module): python wrapper for conv_transpose3d
GitOrigin-RevId: 61097b8713
|
4 years ago |
Megvii Engine Team
|
282dfc6232
|
refactor(imperative): alloc enum type class on heap
GitOrigin-RevId: d2b2acea22
|
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
|
c992afa960
|
test(mge/module): add module reset attribute test
GitOrigin-RevId: 6c9adc4a70
|
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
|
dcff115ef8
|
fix(distributed/launcher): fetch data early to avoid dead lock in launcher
GitOrigin-RevId: 9abcc956ef
|
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
|
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 |