Megvii Engine Team
396412e790
test(mgb/cambricon): add testcase for magicmind runtime opr
GitOrigin-RevId: fa13e842b3
4 years ago
Megvii Engine Team
b5284796f9
feat(mgb/cambricon): add magicmind runtime opr
GitOrigin-RevId: 51af7508de
4 years ago
Megvii Engine Team
623477fdeb
refactor(megbrain): combine mc40 and mc20 compnode and change mc20 to ax
GitOrigin-RevId: bcef8c6f04
4 years ago
Megvii Engine Team
11556cba13
feat(megbrain): add mc20 bazel
GitOrigin-RevId: 96d57656e4
4 years ago
Megvii Engine Team
b8f810ee31
feat(megbrain): add mc20 runtime opr
GitOrigin-RevId: 2ee07b213b
4 years ago
Megvii Engine Team
67e4e83419
fix(lite): fix the force_output_use_user_specified_memory when out var not supported
GitOrigin-RevId: ffaf4c1416
4 years ago
Megvii Engine Team
accb2d8d47
fix(mgb/serialize): fix flatbuffer compatibility issues
GitOrigin-RevId: e4771d6bc4
4 years ago
Megvii Engine Team
5b1383e06c
fix(mgb/plugin): fix io dump for qint4, quint4 and bool type tensors
GitOrigin-RevId: bad1e88690
4 years ago
Megvii Engine Team
24f12df95b
chore(release): bump version
GitOrigin-RevId: 6903f9a962
4 years ago
Megvii Engine Team
605836a70f
fix(mgb): fix trt5.1 test can not create FlattenConcatPlugin
GitOrigin-RevId: 1b9a80b04e
4 years ago
Megvii Engine Team
11d75fecb5
feat(dnn/check_non_finite): add batch check_non_finite
GitOrigin-RevId: e108133282
4 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
4 years ago
Megvii Engine Team
b6c0e8d093
fix(mgb/gopt): update profiler cache of global layout transform test
GitOrigin-RevId: fe8a712599
4 years ago
Megvii Engine Team
f2db7b0ded
feat(mgb/gopt): global layout transform support cuda fp16
GitOrigin-RevId: 1449c54215
4 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
4 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
4 years ago
Megvii Engine Team
fe93013a6e
feat(mgb/gopt): global layout transform support nchw_nchwxx hybrid mode
GitOrigin-RevId: 6d5b55d7fc
4 years ago
Megvii Engine Team
3d45d35241
feat(mgb/gopt): profiler support checking algo availability
GitOrigin-RevId: 39cad612cc
4 years ago
Megvii Engine Team
5f15f75984
test(mgb/gopt): add a testcase for SubGraphExtractor with multiple outputs
GitOrigin-RevId: 7785bdc8c0
4 years ago
Megvii Engine Team
a6230ba95a
feat(mgb/gopt): global layout transform support arm
GitOrigin-RevId: db50b33c11
4 years ago
Megvii Engine Team
0be6ca8899
fix(src/core): fix record change ptr bug on comp node copy
GitOrigin-RevId: 0f68966211
4 years ago
Megvii Engine Team
84baf3df1b
feat(mgb): add tensorrt plugin support
GitOrigin-RevId: 5428b4f665
4 years ago
Megvii Engine Team
b59e8ccf24
fix(mgb): fix cambricon bangc copybara
GitOrigin-RevId: 22b57364e8
4 years ago
Megvii Engine Team
fe6299c52d
Revert "feat(mgb): allow output tensor's ptr change when record"
This reverts commit c610c8bf9a .
GitOrigin-RevId: 2eb0504c5d
4 years ago
Megvii Engine Team
d90cb7763c
feat(src/core): record support change ptr basic
GitOrigin-RevId: b12f1c4a66
4 years ago
Megvii Engine Team
68cdabd288
feat(opr): indexing_multi_axis_vec support nd index
GitOrigin-RevId: 07b1248bdc
4 years ago
Megvii Engine Team
0a56a5b110
feat(cuda/comp_node): gets (maximum) GPU memory allocated/reserved
GitOrigin-RevId: da2cc22436
4 years ago
Megvii Engine Team
13c7c572df
fix(mgb): fix shape infer's condition in lite
GitOrigin-RevId: 550eaff4cd
4 years ago
Megvii Engine Team
0ddabb06cc
fix(imperative/astype): fix astype when target dtype is int4
GitOrigin-RevId: 1c5195176d
4 years ago
Megvii Engine Team
34773ba37b
fix(mgb/gopt): tensorcore pass replace BatchConvBias inputs to nchw4
GitOrigin-RevId: 3ff3c422fb
4 years ago
Megvii Engine Team
481007812e
fix(mgb): support rc version
GitOrigin-RevId: aa5fe5198d
4 years ago
Megvii Engine Team
f00741dad5
feat(build): bazel build support flatbuffer serialization
GitOrigin-RevId: 4755c27e06
4 years ago
Megvii Engine Team
8b7649345a
feat(lite): lite support output var no copy option
GitOrigin-RevId: 5b9488cb93
4 years ago
Megvii Engine Team
7642f66df7
feat(core): support output var no copy
GitOrigin-RevId: fd68a1735b
4 years ago
Megvii Engine Team
7fa5f6f4e2
fix(mgb): correct version of fbs serialization
GitOrigin-RevId: 2a71d9afc2
4 years ago
Megvii Engine Team
25ec2530ba
feat(whl/api/lar): enable megengine dll on Windows
1: reduce python whl package size
2: unify api link logic on all OS
3: add option: MGE_WINDOWS_BUILD_WITH_STATIC_CRT
--- default OFF
--- if build CRT(vc runtime) with STATIC with megengine.dll
some CRT api will crash, for example, flush, so if you
build with static megengine, and do not want to install CRT
you can set MGE_WINDOWS_BUILD_WITH_STATIC_CRT TRUE
--- how to install CRT:
https://docs.microsoft.com/en-us/cpp/windows/latest-supported-vc-redist?view=msvc-160
install VC_redist.x64.exe
4: rename megengine_export to megengine_shared(only export needed symbols ),
caused by runtime symbols conflict with pytorch
GitOrigin-RevId: 93d8d80f29
4 years ago
Megvii Engine Team
c17d603847
feat(mgb): allow output tensor's ptr change when record
GitOrigin-RevId: c610c8bf9a
4 years ago
Megvii Engine Team
d9a9d9d49e
fix(imperative/fastrun): set workspace limit for imperative rt
GitOrigin-RevId: 474dc691a3
4 years ago
Megvii Engine Team
a09a2b730d
fix(mgb/opr): fix fastrun workspace limit for imperative rt
GitOrigin-RevId: bd69a82d4c
4 years ago
Megvii Engine Team
5207154e5c
Revert "fix(mgb): fix fastrun no_profiling_on_shape_change"
This reverts commit 2735536424 .
GitOrigin-RevId: 6a641808f0
4 years ago
Megvii Engine Team
bd42cb06cf
refactor(mgb/lite): refactor lite InfilePersistentCache with core impl
GitOrigin-RevId: 64b7825c34
4 years ago
Megvii Engine Team
d898838e18
fix(src/gopt/test): remove device info in the cache_data
GitOrigin-RevId: 8e9758971d
4 years ago
Megvii Engine Team
a4ac5e7e8f
fix(mgb/gopt): fix profiler cache when build without opr type info name
GitOrigin-RevId: 0eba678f2b
4 years ago
Megvii Engine Team
012de7695d
feat(mgb/gopt): add profiler cache
In order to improve performance of the profiling procedure. Make layout transform testcase stable. The profiling result in ci environment will be cached in files.
GitOrigin-RevId: ba2743f35f
4 years ago
Megvii Engine Team
b33ec46e54
fix(mgb/gopt): fix layout gtrans when graph partition has opr with different format
GitOrigin-RevId: 326fdebb0c
4 years ago
Megvii Engine Team
5885b137fa
feat(dnn/arm): support layout like NHWC channel like broadcast on arm
GitOrigin-RevId: fb4300004c
4 years ago
Megvii Engine Team
565466c25f
feat(lite): auto deduce output tensor shape before model forward
GitOrigin-RevId: 78e00dab5d
4 years ago
Megvii Engine Team
a29f1c8c02
fix(cmake/midout): fix cmake midout
GitOrigin-RevId: 0fb93deb19
4 years ago
Megvii Engine Team
369c2ccc5a
style(all): reformat c++ code
GitOrigin-RevId: 3ffd1b211f
4 years ago
Megvii Engine Team
bfb30dcb81
chore(format): fix compile bugs after code format
GitOrigin-RevId: 11a4b06f6f
4 years ago