Megvii Engine Team
4755400eda
refactor(mge/quantization): add `narrow_range` to control quant dtype's lower bound
GitOrigin-RevId: 92389341be
5 years ago
Megvii Engine Team
c8a9094bb3
fix(mge/data/dataloader): add refcount in _PlasmaStoreManager
GitOrigin-RevId: 9a95cb1a5d
5 years ago
Xu Xinran
05682707fc
chore(version): bump version.
5 years ago
Megvii Engine Team
5257991e68
fix(jit): fix jit doc and add NCHW44_DOT
GitOrigin-RevId: 5f5feae8e7
5 years ago
Megvii Engine Team
cdf25c4afb
feat(mge/utils): export _internal.plugin to mge.utils
GitOrigin-RevId: 795821fe7f
5 years ago
Megvii Engine Team
fda9599a84
feat(mge/quant): add TQT quant method
GitOrigin-RevId: 00b1616e73
5 years ago
Megvii Engine Team
786afef461
feat(build): install CMake config module and pkg-config descriptor
Also, upgrade to CMake 3.13.
The commit also contains significant refactors, as otherwise it is not
possible to properly export target `megengine` to
MegEngine-targets.cmake:
1. Optionally use system provided Flatbuffers.
2. Optionally use system provided MKL-DNN (Tested with Debian).
3. Refactor megbrain and megdnn targets into object libraries.
4. Set different path in BUILD_INTERFACE and INSTALL_INTERFACE of
various target_include_directories.
5. Specify PUBLIC/PRIVATE on various target_link_libraries.
GitOrigin-RevId: df118a879e
5 years ago
Megvii Engine Team
cc4e1dfd7c
feat(mgb/compnode): expose mem_status and try_coalesce_all_free_memory to python
GitOrigin-RevId: 0ec2d556b1
5 years ago
Megvii Engine Team
7aeb4f6ca7
fix(mge/optimizer): use static key to avoid mem leak
GitOrigin-RevId: 85298084a3
5 years ago
Megvii Engine Team
ca52a93e9f
fix(mge/quant): fix init value of histogram observer
GitOrigin-RevId: 9c8caa5f8f
5 years ago
Megvii Engine Team
38f7cbd9aa
fix(mge/module): fix redundant recursion in `train()`
GitOrigin-RevId: 6b3566930b
5 years ago
Megvii Engine Team
5c2323529d
test(mge/quantization): add `quantize_disabled` related test
GitOrigin-RevId: f62ba600c5
5 years ago
Megvii Engine Team
ab91302515
feat(mge/quantization): add `quantize_disabled` attribute in Module
GitOrigin-RevId: f108f03c5a
5 years ago
Megvii Engine Team
08ac685e3a
feat(mge/functional): add logsumexp
GitOrigin-RevId: e7ef50e9ec
5 years ago
Megvii Engine Team
01092feb9b
feat(mgb): add PackAllReducePass
GitOrigin-RevId: 59c1b45393
5 years ago
Megvii Engine Team
c7e6c658fd
refactor(mge/distribute): use is_root (and rank) in stead of rank and root at collective comm
GitOrigin-RevId: dccdb71553
5 years ago
Megvii Engine Team
b8d8886e35
refactor(mge/tensor): combine Dict and TensorDict
GitOrigin-RevId: 6b6c03c04b
5 years ago
Megvii Engine Team
7751a0676e
docs(mge/tensor): add advanced index related docs
GitOrigin-RevId: 31735ddac4
5 years ago
Megvii Engine Team
6742a58b7e
fix(quant): observer do not use cond_take
GitOrigin-RevId: a954814bcb
5 years ago
Megvii Engine Team
35bc0e1f60
fix(mge/function): do not deeply copy saved tensor in Function
GitOrigin-RevId: 3c89d1ceaa
5 years ago
Megvii Engine Team
62753c4d30
fix(mge/sdk): fix comp_node bug in dump_with_testcast_mge
GitOrigin-RevId: 26a8dc50b8
5 years ago
Megvii Engine Team
adfa468899
fix(mge/functional): fix scatter doctest failed for GPU platform issue
GitOrigin-RevId: b5f92c39dd
5 years ago
Megvii Engine Team
d7bb62cfa1
refactor(mgb): move mm_handler from python module into opr-mm
GitOrigin-RevId: f401ce8603
5 years ago
Megvii Engine Team
84068a6bb1
fix(mge/data): fix typos in voc and objects365
GitOrigin-RevId: 491e607b7e
5 years ago
Megvii Engine Team
d2f5874a52
fix(mge/module): fix non-str key error of dict in module
GitOrigin-RevId: f82cd48230
5 years ago
Megvii Engine Team
30b3d3aa3e
fix(dnn/gopt): add convolution nchw44-dot format gopt
GitOrigin-RevId: e8e1e96379
5 years ago
Megvii Engine Team
6d0d5e5a75
fix(mge/redis_server): fix bug for external connection
GitOrigin-RevId: 0b595712d7
5 years ago
Megvii Engine Team
15d1bd4d1f
fix(mge/quantization): use list rather than generator to modify submodules
GitOrigin-RevId: b5a3851330
5 years ago
Megvii Engine Team
7b2c5a7383
fix(mge/quantization): fix histogram observer load and store issue
GitOrigin-RevId: b0a2b476e4
5 years ago
Megvii Engine Team
e6820b91e8
feat(mge/module): add conv and conv_relu quantization module
GitOrigin-RevId: 9cd668d97b
5 years ago
Megvii Engine Team
40412e2688
feat(mge/module): add sync bn
GitOrigin-RevId: ae71a540d1
5 years ago
Megvii Engine Team
caf1fac251
refactor(mge/quantization): split `QATModule` and refactor convert api
GitOrigin-RevId: 80cfb12d10
5 years ago
Megvii Engine Team
d19a637959
fix(python_module): hold GIL during _get_serialized_dtype
GitOrigin-RevId: 6b7288bd0c
5 years ago
Megvii Engine Team
ff341cb19b
fix(mge/quantization): modify observer api
GitOrigin-RevId: 7b9c22be96
5 years ago
Megvii Engine Team
efb60be289
fix(mge/distributed): fix get_free_ports return value
GitOrigin-RevId: 8a546855c4
5 years ago
Megvii Engine Team
1d7fcecab2
feat(mge/serialization): add map location
GitOrigin-RevId: 4b6d83365b
5 years ago
Megvii Engine Team
bb872965e3
feat(mge/module): add local conv2d module
GitOrigin-RevId: bb5ab94b19
5 years ago
Megvii Engine Team
eee3e5594c
feat(mge/distributed): add multiprocess launcher
GitOrigin-RevId: 7d831d125f
5 years ago
Megvii Engine Team
35d46dbb41
fix(mge/functional): simplify the api of add/remove_axis
GitOrigin-RevId: 2482529704
5 years ago
Megvii Engine Team
f7d8b51648
feat(mge/functional): add smooth l1 loss
GitOrigin-RevId: c1437788d7
5 years ago
Megvii Engine Team
3c49d1d324
feat(mge/functional): add hinge loss
GitOrigin-RevId: 64c89c1f8c
5 years ago
Megvii Engine Team
dd8f3ffc4d
test(distributed): replace python command in subprocess with `sys.executable`
GitOrigin-RevId: 1f4262fd0c
5 years ago
Megvii Engine Team
7aa7a09b6b
feat(mge/optimizer): add optimizer adadelta
GitOrigin-RevId: 244bc0d74a
5 years ago
Megvii Engine Team
205291a3b5
feat(mge/quantization): add histgram observer
GitOrigin-RevId: a9252a6baf
5 years ago
Megvii Engine Team
7c4f1a3851
feat(mge/quantization): add calibration support
GitOrigin-RevId: f16fbba2b7
5 years ago
Megvii Engine Team
5eca4da323
feat(mge/functional): add softplus function
GitOrigin-RevId: 26c1ab7416
5 years ago
Megvii Engine Team
855c49ca7d
feat(mge/module): add linear quantization module
GitOrigin-RevId: d0c96a9411
5 years ago
Megvii Engine Team
90107b6dda
feat(mge/cgtools): add network vistior interface with optional pruning
GitOrigin-RevId: cfa69e3e83
5 years ago
Megvii Engine Team
45e2beead6
feat(mgb/gopt): add nchw4 optpass
GitOrigin-RevId: 551b6b828d
5 years ago
Megvii Engine Team
980ebf2c72
feat(mge/module): add fused conv_bn qat approximate version
GitOrigin-RevId: 1b7284a595
5 years ago