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
|
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
|
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
|
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
|
c8f3742f0e
|
fix(mge/doc): fix wrong doc for megengine doc
GitOrigin-RevId: a38dc8e102
|
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 |
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
|
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 |