Megvii Engine Team
5a1f913435
chore(imperative): remove unnecessary function template
GitOrigin-RevId: 8dd2f8c308
4 years ago
Megvii Engine Team
2de2222e46
feat(dnn/cuda): add cutlass batched gemv kernel for matmul operator
GitOrigin-RevId: 51702c4e79
4 years ago
Megvii Engine Team
973d2a0ac2
feat(dnn/cuda): add cutlass matmul using split k parallel
GitOrigin-RevId: 650209e35f
4 years ago
Megvii Engine Team
03c921f7c4
feat(dnn/cuda): add cutlass matmul impls
GitOrigin-RevId: 619c8c299c
4 years ago
Megvii Engine Team
886e7c6eb6
refactor(mge/tensor): support scalar in detach and grad
GitOrigin-RevId: 38349c8357
4 years ago
Megvii Engine Team
a1ca50c923
feat(mge/quantization): add name for quantized module
GitOrigin-RevId: edefbec7b7
4 years ago
Megvii Engine Team
d0f70a445d
fix(imperative/utils): module parameters' name do not have scope after dumping
GitOrigin-RevId: 1497272294
4 years ago
Megvii Engine Team
b9bbf802bd
fix(mgb/opr): fix string append error
GitOrigin-RevId: 295a941edd
4 years ago
Megvii Engine Team
47bdb4ef1d
fix(mgb/core): add MGB_HAVE_THREAD macro to thread_local
GitOrigin-RevId: cfcb50b041
4 years ago
Megvii Engine Team
5b62acfa01
feat(dnn/armv7): add new matmul strategy k8x8x4
GitOrigin-RevId: 0c6b7fa1b2
4 years ago
Megvii Engine Team
ad87f78a14
chore(imperative): refine tblgen for generating op name
GitOrigin-RevId: f47ceae726
4 years ago
Megvii Engine Team
4b98c72171
fix(mge/sublinear): change no virtual grad warning to log_debug
GitOrigin-RevId: 6b5808f2d4
4 years ago
Megvii Engine Team
51868533c8
fix(mgb/gopt): fix opt pass elementwise operation shape issue at tranform to NCHW4
GitOrigin-RevId: c0c4e3f82e
4 years ago
Megvii Engine Team
0a191f8d48
fix(misc): some fixes to make faster_rcnn run in no-symbolic-trace
GitOrigin-RevId: 0fc40ad9c8
4 years ago
Megvii Engine Team
f95bb7b7c9
perf(mge/profiler): reduce profiler overhead
GitOrigin-RevId: 4e1fff9dc4
4 years ago
Megvii Engine Team
9cc732f82d
fix(opencl): fix opencl search algo negative stride support
GitOrigin-RevId: 0642d1718d
4 years ago
Megvii Engine Team
4f3875eb4f
docs(mge): add missing docstring and fix sphinx build warnings
GitOrigin-RevId: 4ce73cfd80
4 years ago
Megvii Engine Team
98d6ba7e2f
refactor(mge/functional): move matinv from nn to math
GitOrigin-RevId: 75c48ce327
4 years ago
Megvii Engine Team
17323dbd73
feat(dist): collect return values in dist.launcher
GitOrigin-RevId: 519e768ce9
4 years ago
Megvii Engine Team
5b697c714c
chore(imperative): remove unused reference to megbrain_graph
GitOrigin-RevId: 2c31f13bb8
4 years ago
Megvii Engine Team
f0170b9c21
fix(whl): opt whl version
example:
1:export PUBLIC_VERSION_POSTFIX=rc1 for create a rc1 version
2:export PUBLIC_VERSION_POSTFIX=dev20210225 for create daily version
3:export PUBLIC_VERSION_POSTFIX=rc1.dev20210225 and export LOCAL_VERSION=mc40 for create special version
GitOrigin-RevId: 8caa7646d0
4 years ago
Megvii Engine Team
47db29aaa2
fix(mge/module): add kwargs param for all modules
GitOrigin-RevId: 7245e669a7
4 years ago
Megvii Engine Team
6fb19b664b
feat(imperative/src): name operators automatically when tracing
GitOrigin-RevId: ff8eb003c5
4 years ago
Megvii Engine Team
09de5a0725
feat(mgb/serialization): be able to serialize operator names
GitOrigin-RevId: d295abb5da
4 years ago
Megvii Engine Team
bb8f292816
refactor(mge/func): change `conv_execution_strategy` to `execution_strategy`
GitOrigin-RevId: 7aef9935ee
4 years ago
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