Megvii Engine Team
eb0cc59dcf
feat(windows/python_whl/cuda): make windows MORE HAPPY for build whl with cuda
GitOrigin-RevId: 9e7d7d4167
4 years ago
Megvii Engine Team
b44e0549b1
fix(mgb): add method to check local cuda env and refactor the way to get cuda include and lib dir
GitOrigin-RevId: 63f7302ce1
4 years ago
Megvii Engine Team
22853fa20c
feat(mge/quantization): add `mapping` parameter for custom modules
GitOrigin-RevId: a4de4261d0
4 years ago
Megvii Engine Team
a7ff580e54
feat(mge/utils): add net stats to calculate parameters and flops
GitOrigin-RevId: a77f89e24b
4 years ago
Megvii Engine Team
b43f6a2602
fix(mge/quantization): handle empty Observer in QATModule
GitOrigin-RevId: e8a62297bc
4 years ago
Megvii Engine Team
13e8f00a37
feat(mge/module): add forward hook support
GitOrigin-RevId: c0db58df13
4 years ago
Megvii Engine Team
ab9fa48ee7
feat(mge/quantization): make `q_dict` a kwarg rather than an arg
GitOrigin-RevId: 38e3b2bfaf
4 years ago
Megvii Engine Team
7a8183f4e0
fix(mge/quantization): fix enable observer bug
GitOrigin-RevId: 493cf9dea9
4 years ago
Megvii Engine Team
555ecea9bc
feat(mge/quantization): add bias fakequant support
GitOrigin-RevId: a5e953b3fa
4 years ago
Megvii Engine Team
9440842e27
fix(mge/core): fix Tensor deepcopy issue
GitOrigin-RevId: 6bea7970b8
4 years ago
Megvii Engine Team
d4b86b844e
feat(mge/dtype): add int2 lowbit support and example
GitOrigin-RevId: 67c14ac959
4 years ago
Megvii Engine Team
ef239f835f
feat(windows/python_whl): make windows HAPPY for build megbrain python package
GitOrigin-RevId: 92b2c07bf9
4 years ago
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