Megvii Engine Team
85b41a90df
feat(dnn): add checksum opr and test
GitOrigin-RevId: e784a76e0b
4 years ago
Megvii Engine Team
a49f4a66b7
feat(dnn): add indexing_one_hot and indexing_set_one_hot opr
GitOrigin-RevId: c5406c71ff
4 years ago
Megvii Engine Team
1310ad4926
feat(dnn): add memory manager for enflame
GitOrigin-RevId: 72a810d959
4 years ago
Megvii Engine Team
2fd3fa8834
feat(cmake): update for enflame cmake compile
GitOrigin-RevId: 3c3c6b3462
4 years ago
Megvii Engine Team
9f2af2099c
feat(mgb): add enflame comp node
GitOrigin-RevId: 478c8538aa
4 years ago
Megvii Engine Team
15d3b3b941
fix(mgb): fix mgb still profiling matmul even when no-profiling-on-shape-change
GitOrigin-RevId: d24f73193e
4 years ago
Megvii Engine Team
31e4bf2c34
fix(mge/optimizer): fix optimizer's state_dict bug
GitOrigin-RevId: 67fb112fb8
4 years ago
Megvii Engine Team
dc96f6aa49
fix(mge/quantization): fix quantized concat forward problem
GitOrigin-RevId: dc21b340d1
4 years ago
Megvii Engine Team
33da8de12b
build(dnn/cuda): split compilation for cutlass wrapper
GitOrigin-RevId: 6365d5fdbc
4 years ago
Megvii Engine Team
63cc5743ca
fix(windows): remove windows build path warning
GitOrigin-RevId: d6e255b7dc
4 years ago
Megvii Engine Team
48a7d333c3
fix(mge/module): fix auto-naming error when there are containers in the module
GitOrigin-RevId: 6b8f043321
4 years ago
Megvii Engine Team
68f2e59763
fix(mgb(ci)): fix tx1 ci testcase
GitOrigin-RevId: 7ac0c4b68d
4 years ago
Megvii Engine Team
420672beca
fix(mgb/dnn): fix x86 matmul midout decl
GitOrigin-RevId: fe1fc977e1
4 years ago
Megvii Engine Team
988f897ee2
test(mge): make test stable
GitOrigin-RevId: 5de478f8b1
4 years ago
Megvii Engine Team
bcbfbbd743
fix(mgb): fix TensorRT runtime opr profiling
GitOrigin-RevId: 3545aa53b2
4 years ago
Megvii Engine Team
702ed9eee6
feat(mge): enable cg dump for debugging
GitOrigin-RevId: 4292364d0d
4 years ago
Megvii Engine Team
ef18a640bc
fix(mgb/core): fix allocate_task_block_unsafe in thread_impl_1.h
GitOrigin-RevId: 21b92fafc5
4 years ago
Megvii Engine Team
fa1ca0ea6c
fix(imperative/opr): fix apply_on_var_node for broadcast
GitOrigin-RevId: 686fff4f73
4 years ago
Megvii Engine Team
e1c83d8d51
fix(mgb/core): add warning information about const_var_shape when record mode
GitOrigin-RevId: a99f9c4e5d
4 years ago
Megvii Engine Team
fc8b501bb8
refactor(mgb/core): refactor cpu compnode so that default cpu has no ability to record
GitOrigin-RevId: 7de4771476
4 years ago
Megvii Engine Team
b717606989
fix(dnn/cuda): add block size limit for culass gemm algo
GitOrigin-RevId: c0940e4535
4 years ago
Megvii Engine Team
55974e8cf9
feat(log): opt log
* opt log at release mode
* add MGE_OVERRIDE_LOG_LEVEL for runtime debug
//! env to config LogLevel
//! DEBUG = 0, INFO = 1, WARN = 2, ERROR = 3, NO_LOG = 4
//! for example , export MGE_OVERRIDE_LOG_LEVEL=0, means set LogLevel to DEBUG
GitOrigin-RevId: 16cd674c56
4 years ago
Megvii Engine Team
c85eefded4
fix(tree): fix copybara
GitOrigin-RevId: 4047219962
4 years ago
Megvii Engine Team
37c3e0a953
chore(mge): remove old version.py and use version_template
4 years ago
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