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
Megvii Engine Team
859081480f
fix(lar): fix load and run get static memory error
GitOrigin-RevId: e11505f551
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
8d825246e8
fix(sdk/load_and_run): remove docs of dump_with_testcase_with_mge.py
GitOrigin-RevId: 4a1138cb55
3 years ago
Megvii Engine Team
38b7cfdec1
fix(mge/utils): fix module states input is dict or others
GitOrigin-RevId: f9701b6134
3 years ago
Megvii Engine Team
1613135990
fix(imperative): insert to candidate only if pinned is 0
GitOrigin-RevId: 2b849d7fe2
3 years ago
Megvii Engine Team
846cc13ca8
fix(mge/functional): fix default dtype of F.full
GitOrigin-RevId: 5cd34af9bf
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
a1cba6cc27
fix(dnn): fix convbias crash on X86
GitOrigin-RevId: 3b33c83d03
3 years ago
Megvii Engine Team
9b4cd92ba3
fix(mgb/dnn): fix cudnnConvBiasActivation crash on nchw32 int8 with oc > 256
GitOrigin-RevId: 20c0b90575
3 years ago
Megvii Engine Team
34773ba37b
fix(mgb/gopt): tensorcore pass replace BatchConvBias inputs to nchw4
GitOrigin-RevId: 3ff3c422fb
3 years ago
Megvii Engine Team
efe3cfe7de
fix(lite): fix the encrypt key in lite
GitOrigin-RevId: 6021d3c89c
3 years ago
huangxinda
d4a080e1ee
feat(mge/third_party): update MegRay version
3 years ago
MegChai
63bb7c2946
docs(mge/functional): update functional.arange docstring
From GITHUB 386
COPYBARA_INTEGRATE_REVIEW=https://github.com/MegEngine/MegEngine/pull/386 from MegChai:docstring-arange 6702c6a5cd
Closes #386
GitOrigin-RevId: e27cdaefe8
3 years ago
Megvii Engine Team
89ed7ab2ff
test(imperative): speed up dtr test
GitOrigin-RevId: 57f092d729
3 years ago
Megvii Engine Team
a2a09ef990
fix(imperative): release dtr related resources when disable dtr
GitOrigin-RevId: eacfded9de
3 years ago
Megvii Engine Team
2676fb7344
refactor(imperative): remove enable_defrag interface as it is enabled by default
GitOrigin-RevId: de9e7d7f16
3 years ago
Megvii Engine Team
d909950f64
refactor(imperative): remove swap in dtr
GitOrigin-RevId: 5c9e42f74a
3 years ago
Megvii Engine Team
bcf69d8f87
refactor(imperative): correctly apply sqrt sampling for dtr
GitOrigin-RevId: dabd365517
3 years ago