Megvii Engine Team
914af2867c
feat(cmake/arm/cuda): support cmake arm with cuda,
* support arm-host and x86-cross-arm
* fix trt myelib cmake build issue at 'trt copy env'(about 'copy env', Please refs cb92123f)
about x86-cross-arm CUDA env:
1: run ./scripts/cmake-build/create_cuda_build_libs.py to prepare
cuda/cudnn/trt env(download deb package info, Please refs create_cuda_build_libs.py)
2: export TRT_ROOT_DIR=xxxxx which may create by step 1
3: export CUDNN_ROOT_DIR=xxxx which may create by step 1
4: export PATH=xxxx:$PATH xxxx need create by step 1, which nvcc have
relative path dir: ../targets/sbsa-linux/
GitOrigin-RevId: 440c76052aabe5b07a4b64d126e759f919c257a8
4 years ago
Megvii Engine Team
096d340c18
fix(whl): fix whl version, fix create manylinux2014 image
GitOrigin-RevId: 3faa59d945
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
30d6b4f65d
fix(mge): fix scalar parameter change to 1-dim parameter after backward and optimize
GitOrigin-RevId: 1794369a71
4 years ago
Megvii Engine Team
cf27dd642c
fix(cuda): use cudnn8.0.4 as cu111 default libs
GitOrigin-RevId: 721ca73bae
4 years ago
Megvii Engine Team
649e4dd750
test(cuda): fix test for cu111
GitOrigin-RevId: 04fe5eb23f
4 years ago
Megvii Engine Team
c69359d00d
fix(dnn/cuda): disable cudnn conv_bias kernels for NCHW4_NCHW tensor format
due to illegal memory access errors caused by the kernels
GitOrigin-RevId: 584cedeeab
4 years ago
Megvii Engine Team
2e4b9a42f7
fix(mgb/gopt): fix folding conv dimshuffle opt pass
GitOrigin-RevId: 878b7de9de
4 years ago
Megvii Engine Team
d62dabe5f8
fix(mgb): fix matmul model compat in flatbuffer
GitOrigin-RevId: 2effad8e5f
4 years ago
Megvii Engine Team
05692d058f
refactor(functional): move matmul dot svd to math pkg
GitOrigin-RevId: 15eb08bacb
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
193e77d4f7
misc(megdnn): misc
GitOrigin-RevId: b321f9b983
4 years ago
Megvii Engine Team
b6376c83a0
fix(mgb/comp_node): use extra physical device to decide whether reuse existed CompNodeImpl
GitOrigin-RevId: 5dddc68a84
4 years ago
Megvii Engine Team
ccbc676158
Revert "fix(whl): append buildid to wheel version"
This reverts commit 00dabfa147
.
GitOrigin-RevId: 87e11e21be
4 years ago
Megvii Engine Team
0e3a6329ff
build(cuda): support cu111 build
GitOrigin-RevId: b3067ba4d5
4 years ago
Megvii Engine Team
e9db061e45
fix(mgb): fix compiling error for cuda-11.1
GitOrigin-RevId: f63e71afa7
4 years ago
Megvii Engine Team
cd02d7c85e
feat(mge/opr): support deformable conv2d/psroi_pooling
GitOrigin-RevId: 501cadda76
4 years ago
jiakai
227d5b9fdb
Update README and fix file permissions.
4 years ago
Megvii Engine Team
f3378100f9
feat(mgb): enable output dynamic memory alloc
GitOrigin-RevId: c809629034
4 years ago
Megvii Engine Team
e82fa4ec23
fix(gopt): using new_inp for build_chain in DelayBroadcast pass
GitOrigin-RevId: efc6377197
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
cd7090acbb
fix(opencl): enable image on mali(cl2.1)
GitOrigin-RevId: 0c670fba80
4 years ago
Megvii Engine Team
fc8d13cda3
fix(build): fix linkage on ubuntu1604
GitOrigin-RevId: ee2a8360b0
4 years ago
Megvii Engine Team
c51a687cef
chore(mge): update copyright years
GitOrigin-RevId: 46104ac891
4 years ago
Megvii Engine Team
7c09e41f18
refactor(mgb): add circular dependency check
GitOrigin-RevId: 01fdb8684b
4 years ago
Megvii Engine Team
af42ce7e69
fix(megdnn): some fixes of execution policy
GitOrigin-RevId: 920f39bcb6
4 years ago
Megvii Engine Team
7afa422df4
refactor(megdnn): refactor sub opr setter
GitOrigin-RevId: 475afb9c10
4 years ago
Megvii Engine Team
821656aa4b
refactor(megdnn): refactor brute force algo in batched matmul
GitOrigin-RevId: 5c143ab3ac
4 years ago
Megvii Engine Team
08ff62deb6
refactor(megdnn): refactor batched matmul algo in conv bias
GitOrigin-RevId: 64fda611ff
4 years ago
Megvii Engine Team
8773926ef8
refactor(megdnn): refactor matmul algo in conv bias
GitOrigin-RevId: 932f7d6f81
4 years ago
Megvii Engine Team
e4b71bdf64
refactor(megdnn): remove unnessary 1x1 algo
GitOrigin-RevId: 809f9b2bba
4 years ago
Megvii Engine Team
44c8d2d16f
refactor(megdnn): refactor matmul algo in deformable conv
GitOrigin-RevId: 05291baf98
4 years ago
Megvii Engine Team
b04ad06f84
refactor(megdnn): refactor matmul algo in conv backward filter
GitOrigin-RevId: bb65777583
4 years ago
Megvii Engine Team
25089e520e
refactor(megdnn): refactor matmul algo in conv backward data
GitOrigin-RevId: 8de601df6e
4 years ago
Megvii Engine Team
3620a940b5
fix(megdnn): fix test with workspace_limits to zero
GitOrigin-RevId: c4ec323361
4 years ago
Megvii Engine Team
0d720653ac
refactor(megdnn): add default algo for convolution forward
GitOrigin-RevId: a12a7d399a
4 years ago
Megvii Engine Team
659217acd2
refactor(megdnn): refactor bfloat16 convbias to recursive inteface
GitOrigin-RevId: 378194fb7f
4 years ago
Megvii Engine Team
4a1d52c9c6
refactor(megdnn): refactor bfloat16 matmul to recursive inteface
GitOrigin-RevId: 641c508aec
4 years ago
Megvii Engine Team
b8febaf91f
refactor(megdnn): refactor bfloat16 convolutionbackwardfilter to recursive inteface
GitOrigin-RevId: 37c08a5b8b
4 years ago
Megvii Engine Team
f14e0c17e7
feat(mgb): add recursive for fastrun and megdnn test
GitOrigin-RevId: 743846f645
4 years ago
Megvii Engine Team
85fa988348
refactor(dnn): add get_algorithm_from_desc interface
GitOrigin-RevId: 6d211ca167
4 years ago
Megvii Engine Team
43b4d4a430
feat(imperative/test): add more correctness tests
GitOrigin-RevId: 2ef6027bb1
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
2b8150ab52
fix(dnn): fix bazel build issue for cambricon platform
GitOrigin-RevId: a79e1b6f6a
4 years ago
Megvii Engine Team
3b28af958a
fix(sdk): add disable-optimize-for-inference option
GitOrigin-RevId: 7e6ea13be7
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