Megvii Engine Team
|
0e82b959a1
|
feat(mge/imperative): add sublinear options
GitOrigin-RevId: f0e917f716
|
4 years ago |
Megvii Engine Team
|
16324e3076
|
feat(dnn/cuda): add remap backward
GitOrigin-RevId: 1b1bcf5db3
|
5 years ago |
Megvii Engine Team
|
b9df2db1c5
|
build(third_party): update megray commit id
GitOrigin-RevId: cc753f941b
|
4 years ago |
Megvii Engine Team
|
387867f8ae
|
feat(mge/quantization): add cambricon-quantization-example
GitOrigin-RevId: ed578ca92b
|
4 years ago |
Megvii Engine Team
|
6e882c1a86
|
feat(whl/imperative): compat for build python whl imperative and legacy runtime
GitOrigin-RevId: 7f6629ae1f
|
4 years ago |
Megvii Engine Team
|
7f857bd471
|
feat(mgb/rocm): add cmake for rocm and fix compile errors and bn
GitOrigin-RevId: c73ed4adc3
|
4 years ago |
Megvii Engine Team
|
3dbac4f47f
|
feat(mge): add atlas_subgraph module
GitOrigin-RevId: 11530383c0
|
4 years ago |
Megvii Engine Team
|
00ef677249
|
fix(mgb): remove internal for cambricon and atlas
GitOrigin-RevId: 861e349eb4
|
4 years ago |
Megvii Engine Team
|
e13fb7b99b
|
fix(mge/quantization): fix apply observer bug which get qparam before call observer
GitOrigin-RevId: 7fd6333683
|
4 years ago |
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 |