Megvii Engine Team
46d4bd8a59
feat(windows): let sdk do not care about more macro on win
GitOrigin-RevId: c522c2fd63
3 years ago
Megvii Engine Team
202b407149
fix(core): fix output var replaced by optpass
GitOrigin-RevId: aea62de345
3 years ago
Megvii Engine Team
e715423f20
feat(src/gopt): add optpass on arm for fusing typecvt and elemwise to elemwise multi type
GitOrigin-RevId: e6bcbbf91b
3 years ago
Megvii Engine Team
d9a46ea47b
fix(dnn): correct behaviour of floor div for int tensor
GitOrigin-RevId: 1444f69cce
3 years ago
Megvii Engine Team
cf1db2616e
fix(fastrun): replace py_redis with cpp_redis to avoid deadlock
GitOrigin-RevId: 9af7fa5c97
3 years ago
Megvii Engine Team
390d2bb545
feat(mgb): tensorrt runtime opr support mutiple profiles
GitOrigin-RevId: 1157d34e4d
3 years ago
Megvii Engine Team
1708ab2ec6
feat(mgb): add tensorrt runtime dynamic batch testcase
GitOrigin-RevId: 36372437ff
3 years ago
Megvii Engine Team
87c845fd61
feat(mgb): tensorrt runtime opr support dynamic batch trt model
GitOrigin-RevId: 7461de704e
3 years ago
Megvii Engine Team
ce119ef5a5
fix(lite): fix lite error when record level is 2
GitOrigin-RevId: 7dabfd8876
3 years ago
Megvii Engine Team
0ad5eeaedd
feat(mgb/gopt): global layout transform support opencl
GitOrigin-RevId: 132605c7d9
3 years ago
Megvii Engine Team
2881934cb8
feat(dnn/check_non_finite): addmul scale to check_non_finite opr
GitOrigin-RevId: c35a219e52
3 years ago
Megvii Engine Team
b8ccc6a211
fix(mgb): fix loss execution policy after opr shallow copy
GitOrigin-RevId: 4738136e4a
3 years ago
Megvii Engine Team
c27f678230
feat(src/opr): add api of training of cpp and related test
GitOrigin-RevId: befb85fd43
3 years ago
Megvii Engine Team
6bb5409976
feat(dnn/src): add images2neibs kernel of opencl and related test
GitOrigin-RevId: 82242b7437
3 years ago
Megvii Engine Team
501eadc1db
fix(mgb): fix copybara mc20
GitOrigin-RevId: 2b491e2278
3 years ago
Megvii Engine Team
2b8e7940b6
fix(lite/cambricon): fix cambricon models which have multiple comp node
GitOrigin-RevId: 624fd7f0ce
3 years ago
Megvii Engine Team
cfad9a5df3
fix(mgb/cambricon): fix magicmind runtime opr when set workspace point second time
GitOrigin-RevId: 1ac9d0eaba
3 years ago
Megvii Engine Team
262e124bec
feat(lite): add lite interface for cambricon models
GitOrigin-RevId: cd5feafd7b
3 years ago
Megvii Engine Team
0c3699ff55
test(imperative/python): fix testcase for magicmind runtime module
GitOrigin-RevId: baf2f72f01
3 years ago
Megvii Engine Team
f398c8e619
feat(imperative): add magicmind runtime opr
GitOrigin-RevId: 02ddb886d9
3 years ago
Megvii Engine Team
396412e790
test(mgb/cambricon): add testcase for magicmind runtime opr
GitOrigin-RevId: fa13e842b3
3 years ago
Megvii Engine Team
b5284796f9
feat(mgb/cambricon): add magicmind runtime opr
GitOrigin-RevId: 51af7508de
3 years ago
Megvii Engine Team
623477fdeb
refactor(megbrain): combine mc40 and mc20 compnode and change mc20 to ax
GitOrigin-RevId: bcef8c6f04
3 years ago
Megvii Engine Team
11556cba13
feat(megbrain): add mc20 bazel
GitOrigin-RevId: 96d57656e4
3 years ago
Megvii Engine Team
b8f810ee31
feat(megbrain): add mc20 runtime opr
GitOrigin-RevId: 2ee07b213b
3 years ago
Megvii Engine Team
67e4e83419
fix(lite): fix the force_output_use_user_specified_memory when out var not supported
GitOrigin-RevId: ffaf4c1416
3 years ago
Megvii Engine Team
accb2d8d47
fix(mgb/serialize): fix flatbuffer compatibility issues
GitOrigin-RevId: e4771d6bc4
3 years ago
Megvii Engine Team
5b1383e06c
fix(mgb/plugin): fix io dump for qint4, quint4 and bool type tensors
GitOrigin-RevId: bad1e88690
3 years ago
Megvii Engine Team
24f12df95b
chore(release): bump version
GitOrigin-RevId: 6903f9a962
3 years ago
Megvii Engine Team
605836a70f
fix(mgb): fix trt5.1 test can not create FlattenConcatPlugin
GitOrigin-RevId: 1b9a80b04e
3 years ago
Megvii Engine Team
11d75fecb5
feat(dnn/check_non_finite): add batch check_non_finite
GitOrigin-RevId: e108133282
3 years ago
Megvii Engine Team
30976c239f
fix(mgb/gopt): fix global layout transform
add a special opr_format modify function for concat operators to modify concat axis when input's layout has been changed
GitOrigin-RevId: 4094208057
3 years ago
Megvii Engine Team
b6c0e8d093
fix(mgb/gopt): update profiler cache of global layout transform test
GitOrigin-RevId: fe8a712599
3 years ago
Megvii Engine Team
f2db7b0ded
feat(mgb/gopt): global layout transform support cuda fp16
GitOrigin-RevId: 1449c54215
3 years ago
Megvii Engine Team
ca7cec7a5d
fix(mgb/gopt): minor fixes for global layout transform
Merge Target::ARM and Target::X86 into Target::CPU to make global layout transform easier to use
GitOrigin-RevId: cc9363fa38
3 years ago
Megvii Engine Team
93152dfa14
fix(mgb/gopt): fix global layout transform
deduplicate the states of the DP problem due to different layout config(NCHW44 & NCHW44_HYBRID) will produce tensors with same layout
GitOrigin-RevId: 7f77efd21b
3 years ago
Megvii Engine Team
fe93013a6e
feat(mgb/gopt): global layout transform support nchw_nchwxx hybrid mode
GitOrigin-RevId: 6d5b55d7fc
3 years ago
Megvii Engine Team
3d45d35241
feat(mgb/gopt): profiler support checking algo availability
GitOrigin-RevId: 39cad612cc
3 years ago
Megvii Engine Team
5f15f75984
test(mgb/gopt): add a testcase for SubGraphExtractor with multiple outputs
GitOrigin-RevId: 7785bdc8c0
3 years ago
Megvii Engine Team
a6230ba95a
feat(mgb/gopt): global layout transform support arm
GitOrigin-RevId: db50b33c11
3 years ago
Megvii Engine Team
0be6ca8899
fix(src/core): fix record change ptr bug on comp node copy
GitOrigin-RevId: 0f68966211
3 years ago
Megvii Engine Team
84baf3df1b
feat(mgb): add tensorrt plugin support
GitOrigin-RevId: 5428b4f665
3 years ago
Megvii Engine Team
b59e8ccf24
fix(mgb): fix cambricon bangc copybara
GitOrigin-RevId: 22b57364e8
3 years ago
Megvii Engine Team
fe6299c52d
Revert "feat(mgb): allow output tensor's ptr change when record"
This reverts commit c610c8bf9a
.
GitOrigin-RevId: 2eb0504c5d
3 years ago
Megvii Engine Team
d90cb7763c
feat(src/core): record support change ptr basic
GitOrigin-RevId: b12f1c4a66
3 years ago
Megvii Engine Team
68cdabd288
feat(opr): indexing_multi_axis_vec support nd index
GitOrigin-RevId: 07b1248bdc
3 years ago
Megvii Engine Team
0a56a5b110
feat(cuda/comp_node): gets (maximum) GPU memory allocated/reserved
GitOrigin-RevId: da2cc22436
3 years ago
Megvii Engine Team
13c7c572df
fix(mgb): fix shape infer's condition in lite
GitOrigin-RevId: 550eaff4cd
3 years ago
Megvii Engine Team
0ddabb06cc
fix(imperative/astype): fix astype when target dtype is int4
GitOrigin-RevId: 1c5195176d
3 years ago
Megvii Engine Team
34773ba37b
fix(mgb/gopt): tensorcore pass replace BatchConvBias inputs to nchw4
GitOrigin-RevId: 3ff3c422fb
3 years ago