Megvii Engine Team
4f86b2c503
fix(mge/subgraph): fix mem leak
GitOrigin-RevId: dc7bfe6265
3 years ago
Megvii Engine Team
cbf9f1367a
fix(imperative): fix CompNodeSyncManager deconstruct
GitOrigin-RevId: 4beb0d5522
3 years ago
Megvii Engine Team
618c77c0b5
feat(dnn): enable eye to support bool
GitOrigin-RevId: 76d874d5b7
3 years ago
Megvii Engine Team
a8941af0c6
fix(docs): fix traced module docs warning
GitOrigin-RevId: a5b790c7b5
3 years ago
Megvii Engine Team
fefd4f0928
fix(traced_module): fix tensor node init
GitOrigin-RevId: bdd8259ce7
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
7a023c059a
feat(mge/traced_module): add optimization api
GitOrigin-RevId: eaa7402640
3 years ago
Megvii Engine Team
6c692b26a7
feat(mge/traced_module): add some fuse passes
GitOrigin-RevId: 065f9df32e
3 years ago
Megvii Engine Team
b28ad4e884
feat(mge/traced_module): add pattern match for TracedModule
GitOrigin-RevId: 0af7b076e6
3 years ago
Megvii Engine Team
2318ea3f15
fix(dnn): fix naive average pooling overflow bug for int8 type
GitOrigin-RevId: b60a7b6cf8
3 years ago
Megvii Engine Team
f64a2e02c9
Revert "fix(mge/subgraph): fix mem leak"
This reverts commit 48beb625e1
.
GitOrigin-RevId: 5fc58d5903
3 years ago
Megvii Engine Team
ea3eb66e7f
docs(mge): modify some docstring about conv and tools
GitOrigin-RevId: 4833e78d71
3 years ago
Megvii Engine Team
acd835d0df
docs(mge): add some docs for cvt_color and GraphInference
GitOrigin-RevId: d78628bfcf
3 years ago
Megvii Engine Team
0185c1a9b9
feat(mge/traced_module): add argspec for top TracedModule
GitOrigin-RevId: 8e31a00c7e
3 years ago
Megvii Engine Team
1daeba76fd
fix(mge/traced_module): fix bn opdef loader
GitOrigin-RevId: d73f953ef6
3 years ago
Megvii Engine Team
2d20f93777
fix(mge/traced_module): fix merge GetAttr failure when qualname is incorrect
GitOrigin-RevId: 46241e5b4f
3 years ago
Megvii Engine Team
e6c271ae46
fix(mge/traced_module): fix some bugs for graph surgery
GitOrigin-RevId: 6328a84cbc
3 years ago
Megvii Engine Team
2d54ad185b
feat(lite): add global layout transform interface for load and run
GitOrigin-RevId: 65c2430ec2
3 years ago
Megvii Engine Team
ba2f0c2e48
fix(dnn/cuda): fix cudnn_conv algo of conv_bias opr for fp16 add z cases
GitOrigin-RevId: b29b009de0
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
3116e128c5
fix(ci/integration_test): fix benchmark torch version
GitOrigin-RevId: bd964ed505
3 years ago
Megvii Engine Team
66f1358666
fix(mge/network): add warning when the node is not in the output list
GitOrigin-RevId: 3240546139
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
c85631aa77
feat(dnn): use ref ptr interface for all backends
GitOrigin-RevId: f65feae5cc
3 years ago
Megvii Engine Team
d90cb7763c
feat(src/core): record support change ptr basic
GitOrigin-RevId: b12f1c4a66
3 years ago
Megvii Engine Team
7b19bc76fb
feat(traced_module): support traced module backward compatible serialization
GitOrigin-RevId: aaa9e51c74
3 years ago
Megvii Engine Team
ffbfe59c9f
fix(mge/subgraph): fix mem leak
GitOrigin-RevId: 48beb625e1
3 years ago
Megvii Engine Team
89186edc5d
fix(dnn): correct reduce/argmxx/fakequant calculation with nan
GitOrigin-RevId: 7e78bdae91
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
05ee603843
fix(mge/distributed): fix single_machine_cache error
GitOrigin-RevId: be57111408
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
896a6fb027
fix(lite/load_and_run): fix lite load_and_run disable_assert_throw
GitOrigin-RevId: abdc0136b6
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
f2f3356572
fix(mge): fix some minor problems
GitOrigin-RevId: 43abda2ab9
3 years ago
Megvii Engine Team
7591718d20
feat(mge): add functional test
GitOrigin-RevId: aa0be62686
3 years ago
Megvii Engine Team
8a222d2c9c
fix(mge): replace functional paramter
GitOrigin-RevId: af3c68c588
3 years ago
Megvii Engine Team
27aa648ba7
fix(mge): control compute_mode by context
GitOrigin-RevId: f3968309f4
3 years ago
huangxinda
dfb9d9801b
feat(ci): remove py format checking
3 years ago