Megvii Engine Team
80f0064365
fix(cuda): show more when cuda env check fail
add cublas version check when build whl
GitOrigin-RevId: 6fcb0269e8
4 years ago
Megvii Engine Team
89e52281e4
fix(build/lite): fix lite rc4_encryptor build failed at new ld env
when cuda enable:
error log:
nn-8.0.4-trt-7.2.2.3-libs/TensorRT-7.2.2.3/lib/libnvinfer_static.a(profile.o):
undefined reference to symbol 'dlclose@@GLIBC_2.2.5'
/usr/bin/ld:
/usr/lib/gcc/x86_64-linux-gnu/9/../../../x86_64-linux-gnu/libdl.so:
error adding symbols: DSO missing from command line
ld version:
GNU ld (GNU Binutils for Ubuntu) 2.34, example ubuntu20.04
error reason: at new ld, import flags with active zone
libnvinfer_static.a depend dlclose@@GLIBC_2.2.5, which need -ldl
ld -o rc4_encryptor xxx.o -ldl libnvinfer_static.a, which failed
so we put a -ldl for CUDA_LIBS env, then will change to
ld -o rc4_encryptor xxx.o libnvinfer_static.a -ldl
GitOrigin-RevId: adda3acb2d
4 years ago
Megvii Engine Team
992a90bbad
docs(mge/quantization): add docstring for Observer
GitOrigin-RevId: 043be3886d
4 years ago
Megvii Engine Team
c45f1eb298
fix(mge/tools): improve `module_visualize` result's robustness and beauty
GitOrigin-RevId: ef7b573776
4 years ago
Megvii Engine Team
786c36ff42
fix(mge/tools): rename `net_stats` in function and examples to match file name
GitOrigin-RevId: 82a1377d66
4 years ago
Megvii Engine Team
9c90ce8c33
fix(mge/quantization): fix `quantize` and `quantize_qat`'s `set_expand_structure` arguments
GitOrigin-RevId: c61633095d
4 years ago
Megvii Engine Team
c8697a7005
feat(imperative/src): python wrapper for cambricon and atlas runtime opr
GitOrigin-RevId: bd969d1339
4 years ago
Megvii Engine Team
8cfed4a1cb
refactor(functional/nn): remove nn.svd
GitOrigin-RevId: 4f58e10a1f
4 years ago
Megvii Engine Team
e6b06914f7
feat(mge/tensor): support non-Tensor value in `_reset` and remove depreciated tests
GitOrigin-RevId: faf6c78aa8
4 years ago
Megvii Engine Team
5a38ad3920
feat(mge/utils): add get/set_expand_structure to deal with complex key
GitOrigin-RevId: 4d1b952068
4 years ago
Megvii Engine Team
fad5bc74c3
feat(mge/tools): add load_network_and_run for megengine
GitOrigin-RevId: 7d8ed5af79
4 years ago
Megvii Engine Team
3f2e15f534
fix(mge/logger): fix set_mgb_log_level
GitOrigin-RevId: 6c94c51483
4 years ago
Megvii Engine Team
594fa722bd
feat(mge/graph): add `modify_opr_algo_strategy_inplace` for fast-run
GitOrigin-RevId: 034cf58b2a
4 years ago
Megvii Engine Team
813be72ce9
feat(mge/utils): refactor GraphInference and add more options
GitOrigin-RevId: 44b96dbf3d
4 years ago
Megvii Engine Team
d970b85deb
feat(functional): add conv3d
GitOrigin-RevId: c06ac77c6c
4 years ago
Megvii Engine Team
46cad4d3c8
feat(functional/ops): add _assert_equal
GitOrigin-RevId: b7ce4158b7
4 years ago
Megvii Engine Team
585aa561f3
fix(functional): replace usage of deprecated get_conv_execution_strategy()
GitOrigin-RevId: 61ef7a50ff
4 years ago
Megvii Engine Team
dcfb6a537e
refactor(mge/functional): move functional api
GitOrigin-RevId: 9cd3e09996
4 years ago
Megvii Engine Team
bfb9def988
test(mge/test): reorganize utils test files
GitOrigin-RevId: 08d27a8b1d
4 years ago
Megvii Engine Team
53075cd3da
feat(mge/experimental): add visualization and net stats for python graph
GitOrigin-RevId: a1ab77c20a
4 years ago
Megvii Engine Team
ae3123b315
feat(mge): add python graph for mgb graph editing
GitOrigin-RevId: 6a9d5beba2
4 years ago
Megvii Engine Team
c82d88751a
fix(dnn/cuda): add cuda nchw int8 conv impl with nchw4 to fix cu111 compatibility
GitOrigin-RevId: 771968f9ac
4 years ago
Megvii Engine Team
0d0968557f
fix(imperative/quantization): add conditional when passing parameters name
GitOrigin-RevId: f38042aeab
4 years ago
Megvii Engine Team
652ec9f251
fix(mgb/dnn): fix backward computation of tqt
GitOrigin-RevId: 850d11a5ce
4 years ago
Megvii Engine Team
2763846168
fix(mge/module): support list/dict/tuple in module __repr__
GitOrigin-RevId: b70193fd79
4 years ago
Megvii Engine Team
07826f5e2b
test(dataloder): fix dataloader multiprocess test
GitOrigin-RevId: c617e13e12
4 years ago
Megvii Engine Team
fe9c6e26a8
feat(imperative/opr): deprecate resize op and make it as a special case of interpolate
GitOrigin-RevId: a5668c5779
4 years ago
Megvii Engine Team
798ae5e58c
fix(imperative/opr): close SCALAR_IDX warning of IndexingMultiAxisVec for proxy_graph
GitOrigin-RevId: 6e0d888f85
4 years ago
Megvii Engine Team
1670a8cf96
docs(mge/quantization): fix wrong unexpected unindent in docstring
GitOrigin-RevId: e604bcebb2
4 years ago
Megvii Engine Team
365e80b599
fix(build): fix cu111 cudnn rpath
GitOrigin-RevId: 11e3227be1
4 years ago
Megvii Engine Team
c89c673594
feat(lite): add windows lite whl scripts, enable megenginelite
pytest and enable cpp test in ci
GitOrigin-RevId: fa963616a6
4 years ago
Megvii Engine Team
4c6484b7f0
feat(lite): add macos lite whl scripts, enable megenginelite
pytest and enable cpp test in ci
GitOrigin-RevId: 5625a514ce
4 years ago
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