Megvii Engine Team
|
7167fdbd49
|
feat(mge/module): add normalization module includes group_norm, instance_norm and layer_norm
GitOrigin-RevId: 9f253e3250
|
4 years ago |
Megvii Engine Team
|
94dba16ff4
|
perf(mge/imperative): misc optimizations
GitOrigin-RevId: bbe7a10b00
|
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
|
6667100638
|
feat(mge): use weakref for GradManger.attach
GitOrigin-RevId: 6df336c3c1
|
4 years ago |
Megvii Engine Team
|
544b798374
|
revert: feat(mge/grad_manager): add `clear_grad` method for GradManager
This reverts commit aa9540e090 .
GitOrigin-RevId: 2d15e503de
|
4 years ago |
Megvii Engine Team
|
176268d20c
|
test(mge): disable gc in leak check
GitOrigin-RevId: b650a46e02
|
4 years ago |
Megvii Engine Team
|
cc85047bf0
|
fix(mge/trace): fix sublinear in trace
GitOrigin-RevId: 356dcd9523
|
4 years ago |
Megvii Engine Team
|
b9918c329d
|
feat(mge/distributed): support distributed key-value store
GitOrigin-RevId: b4abe80014
|
4 years ago |
Megvii Engine Team
|
ab6328c563
|
feat(imperative): port persistent cache
GitOrigin-RevId: 8ca24a37cc
|
4 years ago |
Megvii Engine Team
|
13272eaa63
|
fix(mge/trace): fix random op in symbolic trace
GitOrigin-RevId: 9a851cd177
|
4 years ago |
Megvii Engine Team
|
495472954d
|
fix(trace): link io-op to avoid deadlock
GitOrigin-RevId: 872cb6b715
|
4 years ago |
Megvii Engine Team
|
33e8879af4
|
feat(mge/quantization): support distributed qat
GitOrigin-RevId: c915c843b8
|
4 years ago |
Megvii Engine Team
|
50c4daac85
|
feat(mge/interpreter): add async_level mechanism for Interpreter
GitOrigin-RevId: 8615a23b75
|
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
|
8118a5946d
|
fix(mge/utils): fix get_oprs_seq of cgtools
GitOrigin-RevId: 366a56f4d5
|
4 years ago |
Megvii Engine Team
|
ae8c3c8163
|
feat(mge/functional): add python wrapper for fake quant opr
GitOrigin-RevId: 5f205198be
|
4 years ago |
Megvii Engine Team
|
2e53077915
|
fix(mge/trace): use xpux device when dump
GitOrigin-RevId: f37285f70e
|
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
|
094601e834
|
feat(mge/distributed): allow remote grad by using grad manager
GitOrigin-RevId: a890c206a5
|
4 years ago |
Megvii Engine Team
|
cae8c8a4a4
|
test(mge/parampack): add test for parampack
GitOrigin-RevId: f3e423dffc
|
4 years ago |
Megvii Engine Team
|
28dbadf729
|
feat(imperative): add remap opr
GitOrigin-RevId: 1fa7e18d4b
|
4 years ago |
Megvii Engine Team
|
7cd846a570
|
fix(mge/batch_norm): fix batch_norm check when trace(symbolic=True)
GitOrigin-RevId: 2032eb5f7d
|
4 years ago |
Megvii Engine Team
|
e082e27780
|
refactor(mge/data): Refactor megeninge.data.dataset
GitOrigin-RevId: 1d9c61ce70
|
4 years ago |
Megvii Engine Team
|
be511a56f9
|
fix(mge/imperative): fix tensor astype failed for quantized type
GitOrigin-RevId: 383458acbf
|
4 years ago |
Megvii Engine Team
|
b309890c66
|
docs(mge): pytest for sphinx docstring
GitOrigin-RevId: 8bed12562a
|
4 years ago |
Megvii Engine Team
|
1b56851715
|
fix(mge/trace): fix graph option in trace
GitOrigin-RevId: 7bec84f56d
|
4 years ago |
Megvii Engine Team
|
1217801133
|
perf(mge): add opdef for broadcast
GitOrigin-RevId: 92f0af29eb
|
4 years ago |
Megvii Engine Team
|
fccb2510c4
|
perf(mge): add opdef for elemwise and batchnorm
GitOrigin-RevId: d51fad9867
|
4 years ago |
Megvii Engine Team
|
b327822994
|
feat(mge/grad_manager): add `clear_grad` method for GradManager
GitOrigin-RevId: aa9540e090
|
4 years ago |
Megvii Engine Team
|
2627e1f7df
|
fix(mge/grad_manager): allow multiple calls of `release`
GitOrigin-RevId: 38ca4c78ff
|
4 years ago |
Megvii Engine Team
|
e8bf5bc002
|
fix(mge): ensure contiguous when passing tensor from graph rt to imperative rt
GitOrigin-RevId: e4d944343d
|
4 years ago |
Megvii Engine Team
|
3af1056377
|
feat(mge/grad): attach grad immediately
GitOrigin-RevId: e3a168c03a
|
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
|
786399c5c9
|
fix(pytest/windows/impertive): fix impertive pytest failed on windows
GitOrigin-RevId: 02f4c0a0be
|
4 years ago |
Megvii Engine Team
|
215f88f373
|
fix(dnn/argmxx): fix argmxx on inf
GitOrigin-RevId: 740f67b73a
|
4 years ago |
Megvii Engine Team
|
1fa143ce87
|
refactor(mge/functional): matmul supports symbolic shape, batched mv multiply
GitOrigin-RevId: c4d8cf3306
|
4 years ago |
Megvii Engine Team
|
9748aebeea
|
refactor(mge): tensor_shape -> symbolic_shape
GitOrigin-RevId: 366dc048bf
|
4 years ago |
Megvii Engine Team
|
5cc043f015
|
fix(imperative): sync command queue at exit
GitOrigin-RevId: c0d6aad275
|
4 years ago |
Megvii Engine Team
|
0075e6acb0
|
refactor(mge): refactor cross_entropy
GitOrigin-RevId: 1fac5b5b14
|
4 years ago |
Megvii Engine Team
|
f4b169322d
|
test(mgb/imperative): add adaptive pooling pytest
GitOrigin-RevId: c4dfed1f80
|
4 years ago |
Megvii Engine Team
|
37e56f4b04
|
fix(mge/functional): fix clip under trace(symbolic=True)
GitOrigin-RevId: 5b6f537327
|
4 years ago |
Megvii Engine Team
|
8d1f302590
|
fix(mge): fix batch norm dump
GitOrigin-RevId: eb739437ef
|
4 years ago |
Megvii Engine Team
|
9f4bffbd00
|
fix(mge/tensor): fix valid_broadcast
GitOrigin-RevId: 562b7664e2
|
4 years ago |
Megvii Engine Team
|
af349d6102
|
fix(mge/functional): fix op mismatch when tracing NMSKeep
GitOrigin-RevId: e8f2cbb755
|
4 years ago |
Megvii Engine Team
|
d502e79f85
|
feat(mge): make F.norm numpy compatible
GitOrigin-RevId: 4e5236171c
|
4 years ago |
Megvii Engine Team
|
20e93630f2
|
feat(mge): rename batch_norm2d -> batch_norm
GitOrigin-RevId: 253e8564ea
|
4 years ago |
Megvii Engine Team
|
aa62672629
|
feat(mge): make F.eye numpy compatible
GitOrigin-RevId: fee32537b4
|
4 years ago |
Megvii Engine Team
|
d4e2d27213
|
feat(mge): do not export F.nn.conv_bias_activation
GitOrigin-RevId: e536f87fad
|
4 years ago |
Megvii Engine Team
|
7d8a7e3efe
|
feat(mge): do not export F.loss.*
GitOrigin-RevId: a3ce4d3d76
|
4 years ago |