Megvii Engine Team
|
1edcfa19a8
|
fix(imperative/opr): fix apply_on_var_node for broadcast
GitOrigin-RevId: 686fff4f73
|
4 years ago |
Megvii Engine Team
|
efe264b751
|
test(mge/utils): skip network node convbias test in gpu
GitOrigin-RevId: 9d06cf6c5a
|
4 years ago |
Megvii Engine Team
|
cdbb4a20ff
|
fix(mge/tensor): fix tensor's serialization behavior
GitOrigin-RevId: 4d74a4b46e
|
4 years ago |
Megvii Engine Team
|
9c90ce8c33
|
fix(mge/quantization): fix `quantize` and `quantize_qat`'s `set_expand_structure` arguments
GitOrigin-RevId: c61633095d
|
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
|
3f2e15f534
|
fix(mge/logger): fix set_mgb_log_level
GitOrigin-RevId: 6c94c51483
|
4 years ago |
Megvii Engine Team
|
d970b85deb
|
feat(functional): add conv3d
GitOrigin-RevId: c06ac77c6c
|
4 years ago |
Megvii Engine Team
|
46cad4d3c8
|
feat(functional/ops): add _assert_equal
GitOrigin-RevId: b7ce4158b7
|
4 years ago |
Megvii Engine Team
|
dcfb6a537e
|
refactor(mge/functional): move functional api
GitOrigin-RevId: 9cd3e09996
|
4 years ago |
Megvii Engine Team
|
bfb9def988
|
test(mge/test): reorganize utils test files
GitOrigin-RevId: 08d27a8b1d
|
4 years ago |
Megvii Engine Team
|
ae3123b315
|
feat(mge): add python graph for mgb graph editing
GitOrigin-RevId: 6a9d5beba2
|
4 years ago |
Megvii Engine Team
|
652ec9f251
|
fix(mgb/dnn): fix backward computation of tqt
GitOrigin-RevId: 850d11a5ce
|
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
|
07826f5e2b
|
test(dataloder): fix dataloader multiprocess test
GitOrigin-RevId: c617e13e12
|
4 years ago |
Megvii Engine Team
|
fe9c6e26a8
|
feat(imperative/opr): deprecate resize op and make it as a special case of interpolate
GitOrigin-RevId: a5668c5779
|
4 years ago |
Megvii Engine Team
|
798ae5e58c
|
fix(imperative/opr): close SCALAR_IDX warning of IndexingMultiAxisVec for proxy_graph
GitOrigin-RevId: 6e0d888f85
|
4 years ago |
Megvii Engine Team
|
fa4bf16800
|
feat(mge/functional): add repeat and tile opr
GitOrigin-RevId: a20d4b6fb0
|
4 years ago |
Megvii Engine Team
|
0c37a588ca
|
fix(mge/functional): fix F.ones when input is a tensor of scalar type
GitOrigin-RevId: 6d01d6b58d
|
4 years ago |
Megvii Engine Team
|
1d7dd00144
|
feat(mge/quantization): add QParams and QuantDtypeMeta for quantization data structure
GitOrigin-RevId: df3416fe13
|
4 years ago |
Megvii Engine Team
|
a1ca50c923
|
feat(mge/quantization): add name for quantized module
GitOrigin-RevId: edefbec7b7
|
4 years ago |
Megvii Engine Team
|
d0f70a445d
|
fix(imperative/utils): module parameters' name do not have scope after dumping
GitOrigin-RevId: 1497272294
|
4 years ago |
Megvii Engine Team
|
17323dbd73
|
feat(dist): collect return values in dist.launcher
GitOrigin-RevId: 519e768ce9
|
4 years ago |
Megvii Engine Team
|
6fb19b664b
|
feat(imperative/src): name operators automatically when tracing
GitOrigin-RevId: ff8eb003c5
|
4 years ago |
Megvii Engine Team
|
f211124865
|
feat(mge/trace): support dict return value processing in trace
GitOrigin-RevId: 5b1c08848b
|
4 years ago |
Megvii Engine Team
|
cbff4d7c1a
|
fix(imperative/ops): fix infer_output_attrs_fallible for reshape
GitOrigin-RevId: a93567d79a
|
4 years ago |
Megvii Engine Team
|
d1be31279a
|
fix(mge/quantization): fix tqt load and convert issue and observer calculate params issue
GitOrigin-RevId: f8511f72ad
|
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
|
cd02d7c85e
|
feat(mge/opr): support deformable conv2d/psroi_pooling
GitOrigin-RevId: 501cadda76
|
4 years ago |
Megvii Engine Team
|
a09fc5f784
|
fix(mgb/serialization): disable inplace arith graph opt in graph load
GitOrigin-RevId: d63baf8356
|
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
|
4485e780ae
|
fix(mge/functional): fix linspace device and open other trace tests
GitOrigin-RevId: 4667c4adec
|
4 years ago |
Megvii Engine Team
|
334eda8717
|
refactor(mge): test trace inside opr_test
GitOrigin-RevId: 2cf1135c1c
|
4 years ago |
Megvii Engine Team
|
938152af63
|
fix(mge/functional): convert input type to float32 for more elemwise op
GitOrigin-RevId: cf3bf8cb80
|
4 years ago |
Megvii Engine Team
|
194660462c
|
feat(mge/funcitonal): add cvt_color opr python interface
GitOrigin-RevId: 29e069fb23
|
4 years ago |
Megvii Engine Team
|
ff05667b48
|
test(dist): refactor distributed test with fixtures
GitOrigin-RevId: e69acb72da
|
4 years ago |
Megvii Engine Team
|
e3a3e0cd3c
|
feat(mge/functional): add warp_affine
GitOrigin-RevId: 2b333ccd12
|
4 years ago |
Megvii Engine Team
|
8e50a6daa7
|
feat(mge/utils): add GraphInference in cgtools
GitOrigin-RevId: 72f2201169
|
4 years ago |
Megvii Engine Team
|
364afec033
|
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
|
4 years ago |
Megvii Engine Team
|
4aaae9950b
|
feat(mge/functional): add python wrapper to resize opr
GitOrigin-RevId: b7cc6dd829
|
4 years ago |
Megvii Engine Team
|
df976782fa
|
feat(mge/functional): add matinv
GitOrigin-RevId: d4fa8a8277
|
4 years ago |
Megvii Engine Team
|
5100317660
|
feat(mge): make Function available at autodiff.Function
GitOrigin-RevId: 7c9ab65351
|
4 years ago |
Megvii Engine Team
|
abe3c165ba
|
feat(mge): tensorrt runtime opr
GitOrigin-RevId: 2fdd00adcb
|
4 years ago |
Megvii Engine Team
|
c418d3cd95
|
fix(mge/data): add timeout event
GitOrigin-RevId: 43f2ba1456
|
4 years ago |
Megvii Engine Team
|
0a3ca25337
|
fix(mge): fix backward graph optimization
GitOrigin-RevId: 28fd00ac54
|
4 years ago |
Megvii Engine Team
|
ea8eb4cf72
|
feat(mge/distributed): scalar support for distributed functions
GitOrigin-RevId: 53f3575baf
|
4 years ago |
Megvii Engine Team
|
c522991385
|
test(mge/core): add host compute test, fix async_level and swap/drop test
GitOrigin-RevId: f9b5a83507
|
4 years ago |
Megvii Engine Team
|
7191c4bd9f
|
fix(mge/data): support timeout for serial stream dataloader
GitOrigin-RevId: 1ae5a8cfda
|
4 years ago |
Megvii Engine Team
|
b7ed0cb850
|
fix(mge/quantization): replace `_reset` with "=" in observer
GitOrigin-RevId: ed6af9b98d
|
4 years ago |
Megvii Engine Team
|
0b4918b25e
|
test(mge/quantization): classmethod `from_float_module` of qat module
GitOrigin-RevId: 95c3d45f83
|
4 years ago |