Megvii Engine Team
|
fa4883389a
|
feat(dnn,imperative): remove the restriction of tensor shape when using uint8 region mask
GitOrigin-RevId: 37d9900697
|
2 years ago |
Megvii Engine Team
|
cdf7baa245
|
feat(python): support python 3.10
GitOrigin-RevId: a338ad5c68
|
2 years ago |
Megvii Engine Team
|
8fe5e74f5a
|
Revert "fix(imperative): fix the matrix mul error when symbolic trace"
This reverts commit 3754dc5a65 .
GitOrigin-RevId: 1880b21c10
|
2 years ago |
Megvii Engine Team
|
7f28005847
|
fix(imperative): fix the matrix mul error when symbolic trace
GitOrigin-RevId: 3754dc5a65
|
2 years ago |
Megvii Engine Team
|
68d7320e69
|
fix(mge/utils): fix using wrong function in register_hook_module
GitOrigin-RevId: 75097de1c1
|
2 years ago |
Megvii Engine Team
|
6fac92dd9d
|
feat(whl): force import megbrain before import megengine
GitOrigin-RevId: dc48c56b8a
|
2 years ago |
Megvii Engine Team
|
2694ff81c1
|
fix(mge/module): fix some deconv fuse bn problem
GitOrigin-RevId: e88a633280
|
2 years ago |
Megvii Engine Team
|
2071950297
|
fix(imperative): fix matmul deduce dtype
GitOrigin-RevId: 24f4e1f9fc
|
2 years ago |
Megvii Engine Team
|
582dd4ceb8
|
fix(dnn/sfotmax): call cpu dispatch for softmax opr
GitOrigin-RevId: a606e66101
|
2 years ago |
Megvii Engine Team
|
2976f60bb6
|
fix(test): tmp skip groupnorm test
GitOrigin-RevId: 6d1dd09d47
|
2 years ago |
Megvii Engine Team
|
da7f250c48
|
feat(mge/module): deconv fuse bn and relu
GitOrigin-RevId: 5619b397a4
|
2 years ago |
Megvii Engine Team
|
dbd9483993
|
feat(dnn,src,imperative): add groupnorm op
GitOrigin-RevId: de3c3d10e5
|
2 years ago |
Megvii Engine Team
|
069e4e07a1
|
fix(mge/utils): fix module_stats missing param problem
GitOrigin-RevId: abbf4935da
|
2 years ago |
Megvii Engine Team
|
f444d4fe4d
|
feat(dnn,imperative): region restricted conv support groups=1 even if
depthwise
GitOrigin-RevId: 950d2f4889
|
2 years ago |
Megvii Engine Team
|
d14ae856ce
|
fix(imperative): update region restricted conv testcase for bias shape
GitOrigin-RevId: 8bf66e7312
|
2 years ago |
Megvii Engine Team
|
ffb626243b
|
ci(imperative): fix imperative test_ctc_loss test
GitOrigin-RevId: d48dccc30c
|
2 years ago |
Megvii Engine Team
|
194201a774
|
revert: chore(mge/misc): api converage
This reverts commit --
77fd432cb2
0d7126641b
559d205a8c
6d8e7398e4
b72517bfe9
GitOrigin-RevId: 18dc4ce999
|
2 years ago |
Megvii Engine Team
|
c1c1d6d160
|
feat(imperative): region restrictd conv support bias in python
GitOrigin-RevId: 9a2c1ee27a
|
2 years ago |
Megvii Engine Team
|
9e020d23b4
|
feat(imperative,mgb): add RegionRestrictedConvolution megbrain and imperative opr
GitOrigin-RevId: c0106ade08
|
2 years ago |
Megvii Engine Team
|
edc92ccfd6
|
perf(imperative/data): improve dataloader preformance
GitOrigin-RevId: 7d8d52aaeb
|
2 years ago |
Megvii Engine Team
|
411253f852
|
feat(mge): implement warp_affine backward
GitOrigin-RevId: 7d7261cf64
|
2 years ago |
Megvii Engine Team
|
98460f583f
|
chore(mge/misc): rename APIs that should not be public
GitOrigin-RevId: b72517bfe9
|
2 years ago |
Megvii Engine Team
|
99309fa3d6
|
feat(mge/functional): add param output_padding for deconv ops
GitOrigin-RevId: 8a69608953
|
2 years ago |
Megvii Engine Team
|
bccda5c427
|
fix(mgb/imperative): fix repeat bug in trace mode
GitOrigin-RevId: 9547fc6102
|
2 years ago |
Megvii Engine Team
|
d7c546c90a
|
fix(mge/interpreter): regenerates tensor when its dev value is needed
GitOrigin-RevId: ed26d52ee4
|
2 years ago |
Megvii Engine Team
|
fac67e7c2b
|
feat(gopt): support nchw44 global pooling with fuse_grain
GitOrigin-RevId: 4c43a149f8
|
2 years ago |
Megvii Engine Team
|
d2a1905ad5
|
Revert "feat(mgb): add cumprod opr"
This reverts commit 3436c3bdaa .
GitOrigin-RevId: 95ab3d1aa7
|
2 years ago |
Megvii Engine Team
|
49e14f87b5
|
feat(mgb): add cumprod opr
GitOrigin-RevId: 3436c3bdaa
|
3 years ago |
Megvii Engine Team
|
19af26887e
|
fix(mge/tools): fix module_stats for duplicated module
GitOrigin-RevId: a15f17d616
|
2 years ago |
Megvii Engine Team
|
f386381080
|
fix(imperative): fix inplace operation of optim
GitOrigin-RevId: 2aaa71eb66
|
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
|
9d3977279e
|
fix(traced_module): fix bug of renaming tesnor
GitOrigin-RevId: 468a996bdd
|
3 years ago |
Megvii Engine Team
|
d7ddd43f34
|
feat(imperative): support python3.9 and fix some tests
GitOrigin-RevId: 87e56f9667
|
3 years ago |
Megvii Engine Team
|
cfc41648a4
|
fix(mge): fix grad of maximum(x, x)
GitOrigin-RevId: e0e2efb71b
|
3 years ago |
Megvii Engine Team
|
a49d5bf973
|
fix(autodiff): fix inplace operation on autodiff.Function
GitOrigin-RevId: a658680f35
|
3 years ago |
Megvii Engine Team
|
7252825ce4
|
fix(functional): broadcast_to supports mutable target shape
GitOrigin-RevId: ff79456d5d
|
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
|
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
|
2f829aaa7b
|
test(imperative): speed up integration test
GitOrigin-RevId: b03075fcfc
|
3 years ago |
Megvii Engine Team
|
07bdb3bf1e
|
feat(imperative): add swapaxes
GitOrigin-RevId: e84014a011
|
3 years ago |
Megvii Engine Team
|
e694301721
|
fix(ops/jit): skip lookup include path when nvcc executable not found
GitOrigin-RevId: f5e7dce1c5
|
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
|
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 |