Megvii Engine Team
|
2694ff81c1
|
fix(mge/module): fix some deconv fuse bn problem
GitOrigin-RevId: e88a633280
|
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
|
c2e9860feb
|
chore(license): remove all license in file header
GitOrigin-RevId: a0e31247a6
|
3 years ago |
Megvii Engine Team
|
597a1e791b
|
refactor(imperative): add interface to clear algorithm cache
GitOrigin-RevId: 662618954b
|
3 years ago |
Megvii Engine Team
|
ab6d12caff
|
feat(mge): add conv padding mode
GitOrigin-RevId: 147ced856e
|
3 years ago |
Megvii Engine Team
|
406115dba0
|
fix(imperative): syncbn fp16 support
GitOrigin-RevId: 6059d5b76b
|
3 years ago |
Megvii Engine Team
|
6157d9cfef
|
fix(traced_module): fix Module compatible issue and traced module getattr check
GitOrigin-RevId: 62eb3bfb10
|
3 years ago |
Megvii Engine Team
|
3159eecadd
|
fix(init): fix fan_in and fan_out for group conv2d
GitOrigin-RevId: a6f41063f0
|
3 years ago |
Megvii Engine Team
|
20b42a8c3b
|
fix(dnn): add naive lstm kernel
GitOrigin-RevId: f08ef810cf
|
3 years ago |
Megvii Engine Team
|
2faa6ea5a9
|
Merge pull request #213 from kxz18:rnn
GitOrigin-RevId: 9e9215c115
|
3 years ago |
kxz@thumt102-1
|
8f48da7ffe
|
feat(mgb/opr): add cell level rnn/lstm and sequence level rnn/lstm
|
3 years ago |
Megvii Engine Team
|
5c94db16f2
|
feat(mge/functional): add groups support for conv_transpose2d & 3d
GitOrigin-RevId: b75b792fb4
|
3 years ago |
Megvii Engine Team
|
ac86d64474
|
fix(mge/module): fix prelu error when use_symbolic_shape is true
GitOrigin-RevId: 25b9c4d41d
|
3 years ago |
Megvii Engine Team
|
f7d05db7d6
|
fix(mge/module): fix named_tensors
GitOrigin-RevId: bb5aa1f41d
|
3 years ago |
Megvii Engine Team
|
74cbc10d82
|
feat(opr): let batchnorm support empty IO
GitOrigin-RevId: 219411c80c
|
3 years ago |
Megvii Engine Team
|
e5217db9f9
|
feat(traced_module): move traced_module out of the experimental folder
GitOrigin-RevId: 36c76b5277
|
3 years ago |
Megvii Engine Team
|
442b4f6c26
|
test(traced_module): add some testcases for traced module
GitOrigin-RevId: 0d6bb20b2b
|
4 years ago |
Megvii Engine Team
|
a7a3bf2d6c
|
test(subgraph): simple test for subgraph
GitOrigin-RevId: 3d6ecd5db7
|
3 years ago |
Megvii Engine Team
|
3eb0505f9b
|
feat(imperative): add support for quantized conv transpose2d
GitOrigin-RevId: ffd6431299
|
3 years ago |
Megvii Engine Team
|
3f3a256e0f
|
fix(mge/functional): fix conv* dtype promotion
GitOrigin-RevId: 3f03790cfc
|
3 years ago |
Megvii Engine Team
|
207a346351
|
chore(mge): run get_device_count("gpu") in subprocess
GitOrigin-RevId: 0f0dc001cf
|
4 years ago |
Xinda Huang
|
c90fa087ea
|
test(mge): delete test_external.py
|
3 years ago |
Megvii Engine Team
|
b2944559a8
|
fix(imperative/module): remove ``__getattribute__`` method in module
GitOrigin-RevId: 5ac525f010
|
4 years ago |
Megvii Engine Team
|
1bf18252c4
|
feat(mge/amp): add mix precision autocast support
GitOrigin-RevId: 6fbffc4845
|
4 years ago |
Megvii Engine Team
|
427113088b
|
fix(module/normalization): fix bug of LayerNorm and support input of any shape
GitOrigin-RevId: fd643addb5
|
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
|
7f9d441cbb
|
docs(mge/module): add note about weight shape in conv
GitOrigin-RevId: 43e1f15968
|
4 years ago |
Megvii Engine Team
|
e18afa0b05
|
feat(mge/module): python wrapper for conv_transpose3d
GitOrigin-RevId: 61097b8713
|
4 years ago |
Megvii Engine Team
|
c992afa960
|
test(mge/module): add module reset attribute test
GitOrigin-RevId: 6c9adc4a70
|
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
|
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
|
2763846168
|
fix(mge/module): support list/dict/tuple in module __repr__
GitOrigin-RevId: b70193fd79
|
4 years ago |
Megvii Engine Team
|
02df634da2
|
fix(mge/quantization): set ``q_dict`` as an instance property
GitOrigin-RevId: 2f32008aad
|
4 years ago |
Megvii Engine Team
|
91a3580f75
|
refactor(mge/cgtools): remove load_and_inference and use GraphInference
GitOrigin-RevId: 0e688ebd59
|
4 years ago |
Megvii Engine Team
|
ff05667b48
|
test(dist): refactor distributed test with fixtures
GitOrigin-RevId: e69acb72da
|
4 years ago |
Megvii Engine Team
|
364afec033
|
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
|
4 years ago |
Megvii Engine Team
|
abe3c165ba
|
feat(mge): tensorrt runtime opr
GitOrigin-RevId: 2fdd00adcb
|
4 years ago |
Megvii Engine Team
|
9d928e7f83
|
refactor(mge/distributed): sync interpreter for distribtued launcher
GitOrigin-RevId: 8a88c272a1
|
4 years ago |
Megvii Engine Team
|
ee4ea7fdc8
|
test(distributed/test): make distributed test more stronger
GitOrigin-RevId: 085fd1dcfd
|
4 years ago |
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
|
442058aead
|
feat(mge/quantize): add batch matmul activation module for inference
GitOrigin-RevId: f6dbd1f4c0
|
4 years ago |
Megvii Engine Team
|
33e8879af4
|
feat(mge/quantization): support distributed qat
GitOrigin-RevId: c915c843b8
|
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
|
2627e1f7df
|
fix(mge/grad_manager): allow multiple calls of `release`
GitOrigin-RevId: 38ca4c78ff
|
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
|
9748aebeea
|
refactor(mge): tensor_shape -> symbolic_shape
GitOrigin-RevId: 366dc048bf
|
4 years ago |