Megvii Engine Team
|
d19fc2c1da
|
fix(imperative): add alloc TensorPtr in imperative
GitOrigin-RevId: 1b438fc436
|
3 years ago |
Megvii Engine Team
|
ac51f7807d
|
feat(mge/distributed): add support for batch send recv op
GitOrigin-RevId: eb3d712704
|
3 years ago |
Megvii Engine Team
|
2886245bb1
|
perf(imperative/src): improve pad host performance
GitOrigin-RevId: 05223deca7
|
3 years ago |
Megvii Engine Team
|
5a35513856
|
fix(mgb): fix profile skip condition
GitOrigin-RevId: f196eabc98
|
3 years ago |
Megvii Engine Team
|
5bdc430ec1
|
fix(mgb/fastrun): fix megbrain fastrun memory overflow bug
GitOrigin-RevId: f56aa5a505
|
3 years ago |
Megvii Engine Team
|
2484cd2741
|
fix(tensor): check args when construct tensor with existing tensor
GitOrigin-RevId: 0345454070
|
3 years ago |
Megvii Engine Team
|
d3247bee5a
|
fix(dtr): always write shape when tensor produced
GitOrigin-RevId: d2b23b5c25
|
3 years ago |
Megvii Engine Team
|
1192a9a61b
|
fix(imperative): fix adaptive pool2d
GitOrigin-RevId: 624b47a51a
|
3 years ago |
Megvii Engine Team
|
24c5c19bf0
|
fix(imperative): make functional ops support negative axis
GitOrigin-RevId: f61e01270b
|
3 years ago |
Megvii Engine Team
|
f96429c031
|
feat(imperative): support empty tensor in roi_align
GitOrigin-RevId: aeb2770401
|
3 years ago |
Megvii Engine Team
|
02c1a0c3de
|
fix(imperative/format): warn once when parameter without format been attached
GitOrigin-RevId: 02174447f6
|
3 years ago |
Megvii Engine Team
|
c2deef1a97
|
feat(mge): aad atlas710 support
GitOrigin-RevId: 6458c5c23c
|
3 years ago |
Megvii Engine Team
|
df5ebd3da7
|
fix(imperative/ops): fix the vmemory problem in 1.9
GitOrigin-RevId: 728e983701
|
3 years ago |
Megvii Engine Team
|
8563f51404
|
fix(imperative): fix buildin reduce keepdim
GitOrigin-RevId: 38d90ab38a
|
3 years ago |
Megvii Engine Team
|
da91e650a5
|
refactor(ops/layer_norm): speed up the host speed of layer_norm
GitOrigin-RevId: 6f359b5b29
|
3 years ago |
Megvii Engine Team
|
67cfce9f2f
|
fix(imperative/amp): add is_scalar check in elemwise and concat
GitOrigin-RevId: 61a612e92a
|
3 years ago |
Megvii Engine Team
|
d313f92610
|
fix(imperative/amp): fix format transformation for symbol trans
GitOrigin-RevId: 96cc237c67
|
3 years ago |
Megvii Engine Team
|
261a5bce23
|
feat(imperative/amp): add dimshuffle in set_format for nhwc
GitOrigin-RevId: 5ced9e1a31
|
3 years ago |
Megvii Engine Team
|
c9e56f4987
|
feat(imperative/amp): add dimshuffle before creating nhwc tensor
GitOrigin-RevId: 4461f9a0d3
|
3 years ago |
Megvii Engine Team
|
d57a071271
|
feat(imperative/amp): add fallback for op not supported for nhwc tensor
GitOrigin-RevId: 8411ce7bdc
|
3 years ago |
Megvii Engine Team
|
38a9aa9faf
|
feat(imperative/amp): add auto dimshuffle for elemwise and concat
GitOrigin-RevId: 6e3df4e064
|
3 years ago |
Megvii Engine Team
|
cd26376549
|
style(imperative/amp): reformat code
GitOrigin-RevId: 6e5a6e1eaf
|
3 years ago |
Megvii Engine Team
|
6f0b582064
|
chore(imperative/amp): adapt dev
GitOrigin-RevId: 41eb0faadf
|
3 years ago |
Megvii Engine Team
|
ee984e8608
|
fix(imperative/amp): fix distributed backward callback for nhwc amp
GitOrigin-RevId: 4d725b0ea4
|
3 years ago |
Megvii Engine Team
|
15c6da6218
|
feat(imperative/amp): add nhwc support for adaptive pooling
GitOrigin-RevId: 7c5755308e
|
3 years ago |
Megvii Engine Team
|
c28a875fac
|
fix(imperative/amp): adapt new transformation
GitOrigin-RevId: 6edd577a70
|
3 years ago |
Megvii Engine Team
|
fd41302cc1
|
feat(imperative/amp): add set_format
GitOrigin-RevId: 91de6f49de
|
3 years ago |
Megvii Engine Team
|
fc633ce4ff
|
fix(imperative/amp): fix custom grad in Subgraph
GitOrigin-RevId: 1c728d6ab9
|
3 years ago |
Megvii Engine Team
|
673b295d75
|
feat(imperative/amp): remove conv_format and bn param_dim configs
GitOrigin-RevId: 848d34f63d
|
3 years ago |
Megvii Engine Team
|
7e9aa742e6
|
feat(imperative/amp): enable auto_convert_format by default
GitOrigin-RevId: 71ae311fed
|
3 years ago |
Megvii Engine Team
|
533fb5bf49
|
feat(imperative): support formatted tensor and add special op rules
GitOrigin-RevId: 77ff909f23
|
3 years ago |
Megvii Engine Team
|
4aa79c453b
|
perf(mge): override grad of matmul
GitOrigin-RevId: d9d97e70fe
|
3 years ago |
Megvii Engine Team
|
98b5ee78c1
|
feat(mge/dnn): add lamb optimizer
GitOrigin-RevId: 5a27157456
|
3 years ago |
Megvii Engine Team
|
a926878c01
|
feat(imperative): remove symbolvar of imperative
GitOrigin-RevId: 16da6d1491
|
3 years ago |
Megvii Engine Team
|
c2e9860feb
|
chore(license): remove all license in file header
GitOrigin-RevId: a0e31247a6
|
3 years ago |
Megvii Engine Team
|
1783b8977a
|
feat(profiler): integrate cupti backend
GitOrigin-RevId: dec8be1908
|
3 years ago |
Megvii Engine Team
|
58df717e37
|
fix(mge/autodiff): fix attaching tensor already in gradient path
GitOrigin-RevId: da774509ca
|
3 years ago |
Megvii Engine Team
|
8182af6eb6
|
fix(mgb): fix strategy of grad_op and opr_attr
GitOrigin-RevId: bb7ab8fa9d
|
3 years ago |
Megvii Engine Team
|
e2f5156b69
|
refactor(megbrain): save fastrun result to algorithm cache
GitOrigin-RevId: 45301ebb4d
|
3 years ago |
Megvii Engine Team
|
4b27e861f4
|
fix(ops): implement from_op_node for reshape
GitOrigin-RevId: 4c99438504
|
3 years ago |
Megvii Engine Team
|
4fb3d886b8
|
perf(misc): use reinterpret_cast to convert valueshape
GitOrigin-RevId: ee14c7ce5e
|
3 years ago |
Megvii Engine Team
|
951ed476d6
|
fix(minigraph): supports varnode forwarding
GitOrigin-RevId: 4494106f0a
|
3 years ago |
Megvii Engine Team
|
27d4c4b36c
|
refactor(stats): use static inline variable declaration
GitOrigin-RevId: 7d86e5f257
|
3 years ago |
Megvii Engine Team
|
53d48ca399
|
fix(fastrun): persistent cache add prefix
(cherry picked from commit bdd82c1976 )
GitOrigin-RevId: 0b13d32175
|
3 years ago |
Megvii Engine Team
|
7bf5b0ee1e
|
test(imperative): check env values after each pytest
GitOrigin-RevId: 826788113a
|
3 years ago |
Megvii Engine Team
|
409c988163
|
fix(imperative): add matmul apply_on_varnode
GitOrigin-RevId: 2cf6bf237c
|
3 years ago |
Megvii Engine Team
|
275f12c98a
|
fix(mge): fix dimshuffle shape infer
GitOrigin-RevId: ad7a73fbd1
|
3 years ago |
Megvii Engine Team
|
e59b6e13a3
|
fix(imperative/src): fix empty_tensor bug of convbwd&rng
GitOrigin-RevId: 4c948f41f0
|
3 years ago |
Megvii Engine Team
|
7927e98fd6
|
perf(mge): speed up PixelShuffle
GitOrigin-RevId: 942e755745
|
3 years ago |
Megvii Engine Team
|
f7e10ea8d7
|
perf(imperative): improve matmul/batch_matmul
GitOrigin-RevId: 4ceb2eb601
|
3 years ago |