Megvii Engine Team
ef079b84fb
feat(lite): add linux lite whl scripts, enable megenginelite pytest/cmake gtest in ci
GitOrigin-RevId: 0243029c2d
4 years ago
Megvii Engine Team
3fab21ed31
fix(lite): fix BUILD and misc
GitOrigin-RevId: 48aa0e0924
4 years ago
Megvii Engine Team
65c3159cf1
feat(lite): add version interface
GitOrigin-RevId: cd09b271a9
4 years ago
Megvii Engine Team
af60ff975c
feat(lite): add cmake build for lite
GitOrigin-RevId: 2424eb2681
4 years ago
Megvii Engine Team
1ee331dd91
fix(git): fix ci resource lfs, caused by f5993
GitOrigin-RevId: 12e443512d
4 years ago
Megvii Engine Team
f2b42bf09e
chore(dotprod): add arm dotprod attribute for easy use
GitOrigin-RevId: 78c3e72218
4 years ago
Megvii Engine Team
fa4bf16800
feat(mge/functional): add repeat and tile opr
GitOrigin-RevId: a20d4b6fb0
4 years ago
Megvii Engine Team
c33a717314
feat(dnn): repalce is_reproducible with algo attribute in opencl, cpu, rocm and cuda
GitOrigin-RevId: 86dead0a11
4 years ago
Megvii Engine Team
249a116bec
fix(linux/whl): fix megengine whl build faild at manylinux2014
as > cuda11 cudnn8 trt7 have some dlopen libs, auditwheel can
not handle correctly, also at lastest auditwheel, NEW LIBS dirs
change from package data dir to package dir(eg, megengine to MegEngine),
which lead to copy manually libs change rpath hard with auditwheel
rpath, so do not auditwheel now!
GitOrigin-RevId: 3dfc8a5267
4 years ago
Megvii Engine Team
0c37a588ca
fix(mge/functional): fix F.ones when input is a tensor of scalar type
GitOrigin-RevId: 6d01d6b58d
4 years ago
Megvii Engine Team
b0944dc72a
fix(mgb): fix tx1 compile
GitOrigin-RevId: 0eeb62ff73
4 years ago
Megvii Engine Team
ce88e6c4f7
feat(mge/quantization): use extra act_fakequant to decide whether to do bias fakequant
GitOrigin-RevId: bf54012155
4 years ago
Megvii Engine Team
1d7dd00144
feat(mge/quantization): add QParams and QuantDtypeMeta for quantization data structure
GitOrigin-RevId: df3416fe13
4 years ago
Megvii Engine Team
4130dcd355
fix(mge/quantization): fix QATModule filter in `reset_qconfig` and `hook_qat_module`
GitOrigin-RevId: 92e9f36ca4
4 years ago
Megvii Engine Team
97beae2fd8
fix(megdnn): fix megdnn benchmark testcase
GitOrigin-RevId: 726971474a
4 years ago
Megvii Engine Team
1d4baa4284
fix(imperative/dtype): fix is_dtype_equal
GitOrigin-RevId: b097c9c3ca
4 years ago
Megvii Engine Team
0035efa636
fix(mge): limit task queue size
GitOrigin-RevId: 9481d38914
4 years ago
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