Megvii Engine Team
efe264b751
test(mge/utils): skip network node convbias test in gpu
GitOrigin-RevId: 9d06cf6c5a
4 years ago
Megvii Engine Team
c6069d02d7
chore(scripts): remove manylinux2010 build enviroment
4 years ago
Megvii Engine Team
b730547b4e
chore(third_party): update upstream of cutlass
4 years ago
Megvii Engine Team
e9982d613f
chore(mge/functional): add compatible code for functional api
GitOrigin-RevId: 3b2f829cc5
4 years ago
Megvii Engine Team
8928c77c56
fix(neuropilot): fix loader header mismatch
GitOrigin-RevId: c6b3b7a7b8
4 years ago
Megvii Engine Team
221ec38a48
feat(imperative): reduce profiler overhead
GitOrigin-RevId: dded9d9391
4 years ago
Megvii Engine Team
1c01128ff5
fix SASS cu111
GitOrigin-RevId: cb075ad154
4 years ago
Megvii Engine Team
58c8746e30
fix(opr): fix fast-run error in cuda
GitOrigin-RevId: 28dd187df9
4 years ago
Megvii Engine Team
ba2ad46e54
feat(gopt): add deconv nchw4 int8 opt pass, add deconv nchw int8
GitOrigin-RevId: c0530a949e
4 years ago
Megvii Engine Team
5d350fc843
feat(dnn/cuda): add deconv int8 and fix cutlass conv wrapper base on modify cutlass 2.4
GitOrigin-RevId: 49e0565e8a
4 years ago
Megvii Engine Team
4917534b65
feat(mge/optimizer): save state's numpy value by default in `state_dict`
GitOrigin-RevId: ec7e4d56f5
4 years ago
Megvii Engine Team
84f990a051
fix(mge/tensor): fix `__repr__` for Tensor's subclass
GitOrigin-RevId: 91b58e6dca
4 years ago
Megvii Engine Team
cdbb4a20ff
fix(mge/tensor): fix tensor's serialization behavior
GitOrigin-RevId: 4d74a4b46e
4 years ago
Megvii Engine Team
9da26407ea
feat(ci): and model to test algo policy compatible
GitOrigin-RevId: e58caf08c8
4 years ago
Megvii Engine Team
e19b9af19b
feat(imperative): add bit combined enum to python C extension
GitOrigin-RevId: 92307dd2ca
4 years ago
Megvii Engine Team
a3ea1f153c
feat(mgb/opr): add fast profile and combined Execution strategy
GitOrigin-RevId: 843dc3a790
4 years ago
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