Megvii Engine Team
e258812f12
feat(dnn): add bool dtype
GitOrigin-RevId: 98c8a092b4
4 years ago
Megvii Engine Team
35c712767d
fix(mge/quant): fix TQT epoch scale change bug
GitOrigin-RevId: 6e39de9cec
4 years ago
Megvii Engine Team
e6e41242c7
fix(mge/quant): fix zero grad warn in TQT train
GitOrigin-RevId: a6545ee366
4 years ago
Megvii Engine Team
4a01d23631
ci(mgb/python_module): fix doctest syntax error
GitOrigin-RevId: e98ca810ea
4 years ago
Megvii Engine Team
4495c0cc86
feat(mgb/quantization): add get quantize parameters support
GitOrigin-RevId: 5727f63560
4 years ago
Megvii Engine Team
380cb6e47f
feat(mge/jit): add support output symbol var name settings for dump
GitOrigin-RevId: 258c03ee34
4 years ago
Megvii Engine Team
134a102693
fix(mge/doc): fix typos, format
GitOrigin-RevId: 7d8037434e
4 years ago
Megvii Engine Team
7a8a2830a8
feat(quant): support nnie quant
GitOrigin-RevId: 8ca3f828bd
5 years ago
Megvii Engine Team
0278325055
fix(python/import/macos): fix some macos user import crash
GitOrigin-RevId: b701710b74
4 years ago
Megvii Engine Team
8269215959
fix(mge/module): fix named_children of Sequential
GitOrigin-RevId: d3220fb361
4 years ago
Megvii Engine Team
b689b4ce5c
fix(mge/functional): add F.clamp to __all__ to fix docs
GitOrigin-RevId: 8857e2316b
4 years ago
Megvii Engine Team
177c268898
feat(osx/python_whl): fix python pytest failedon osx
GitOrigin-RevId: 97d2b496bb
4 years ago
Megvii Engine Team
4d541b72ce
feat(osx/python_whl): make macos HAPPY for build megbrain python package
GitOrigin-RevId: 7ffd9b9541
5 years ago
Megvii Engine Team
b2f05bf8d8
fix(mge/module): fix quantized fold weight value range limit for fused conv/bn modules
GitOrigin-RevId: 007c2f13b6
4 years ago
Megvii Engine Team
0df74604bd
fix(data): fix contiguous id
GitOrigin-RevId: 7f79cda0b5
4 years ago
Megvii Engine Team
3bd8ef3589
feat(mgb/compnode): add atlas compnode
GitOrigin-RevId: 19f3c33003
5 years ago
Megvii Engine Team
aa147b7426
fix(osx/python_module): fix build python module on osx
GitOrigin-RevId: 5dde1b4e81
5 years ago
Megvii Engine Team
c8f3742f0e
fix(mge/doc): fix wrong doc for megengine doc
GitOrigin-RevId: a38dc8e102
5 years ago
Megvii Engine Team
cf3a55ce17
fix(mgb/opr-mm): remove PeerDesc from RemoteSend and RemoteRecv
GitOrigin-RevId: b7a7bbd0da
5 years ago
Megvii Engine Team
f856b17085
fix(mge/functional): reshape bias to (1, out_features) in linear
GitOrigin-RevId: a15880b7fc
5 years ago
Megvii Engine Team
caf77d000e
feat(mge/module): add quantize dtype load support for module load_state_dict
GitOrigin-RevId: 0a94cb6b17
5 years ago
Megvii Engine Team
ef437f69aa
fix(mge/data/dataloader): add refcount in _PlasmaStoreManager
GitOrigin-RevId: 9a95cb1a5d
5 years ago
Megvii Engine Team
2b3a88d011
feat(mge/distributed): add parameter replica_mode
GitOrigin-RevId: 244e4ca437
5 years ago
Megvii Engine Team
6972fc7d75
feat(opr/standalone): import NMSKeep from MegSkull contrib
1. clang-format.
2. Use name "NMSKeepMGB" during serialization to avoid conflict with
MegSkull contrib one.
GitOrigin-RevId: 627c60e482
5 years ago
Megvii Engine Team
5fa444c39a
refactor(mge/quantization): add `narrow_range` to control quant dtype's lower bound
GitOrigin-RevId: 92389341be
5 years ago
Megvii Engine Team
6d367454cf
feat(mge/opr-mm): add param local_grad for collective_comm opr
GitOrigin-RevId: cc120cfb55
5 years ago
Megvii Engine Team
4ace67ff44
test(mge/distributed): check gpu num for multi gpu test
GitOrigin-RevId: 78f4718682
5 years ago
Xu Xinran
b532284cc2
chore(version): bump version.
5 years ago
Megvii Engine Team
af3de7e136
fix(jit): fix jit doc and add NCHW44_DOT
GitOrigin-RevId: 5f5feae8e7
5 years ago
Megvii Engine Team
7b8a99517c
feat(mge/utils): export _internal.plugin to mge.utils
GitOrigin-RevId: 795821fe7f
5 years ago
Megvii Engine Team
6ed791bc86
feat(mge/quant): add TQT quant method
GitOrigin-RevId: 00b1616e73
5 years ago
Megvii Engine Team
328fb36f17
feat(mgb/opr-mm): add Scatter, Gather, AllToAll oprs
GitOrigin-RevId: f75169ecd6
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