Megvii Engine Team
|
9a79342d35
|
refactor(mge/virtualdep): remove virtualdep opdef
GitOrigin-RevId: bbe3ae3fa3
|
4 years ago |
Megvii Engine Team
|
88b402ef62
|
feat(mge/trace): tracing use id to set priority
GitOrigin-RevId: 6e1f1ece0e
|
4 years ago |
Megvii Engine Team
|
cde9727a4f
|
fix(mge/quantization): fix `qat.conv_bn` parameter issue
GitOrigin-RevId: 4aa3836926
|
4 years ago |
Megvii Engine Team
|
94796060b6
|
test(mge/quantization): qat module, quantized op and module
GitOrigin-RevId: 17b28060cc
|
4 years ago |
Megvii Engine Team
|
ab9f44f15c
|
feat(mge/quantization): add support for easyquant
GitOrigin-RevId: 060d908349
|
4 years ago |
Megvii Engine Team
|
69e3e32240
|
feat(imperative): auto generated opdef header and python binding
GitOrigin-RevId: d2f22ad5fe
|
4 years ago |
Megvii Engine Team
|
b9c37112a2
|
refactor(mge/distributed): skip barrier when running with single node
GitOrigin-RevId: 6cd5a2e1e2
|
4 years ago |
Megvii Engine Team
|
ee4ea7fdc8
|
test(distributed/test): make distributed test more stronger
GitOrigin-RevId: 085fd1dcfd
|
4 years ago |
Megvii Engine Team
|
3ecded74ea
|
refactor(distributed/server): use port 0 to get available port
GitOrigin-RevId: e367846b92
|
4 years ago |
Megvii Engine Team
|
7aa54b0ec6
|
feat(mge): enable memory swap and drop/recomputation
GitOrigin-RevId: c56c87c88b
|
4 years ago |
Megvii Engine Team
|
2ad8c5e1e9
|
fix(mge/io_remote): fix remote send/recv gradient at trace
GitOrigin-RevId: 7886efd0c1
|
4 years ago |
Megvii Engine Team
|
6c4841e807
|
fix(mge/quantization): `disable_fake_quant` does not work correctly
GitOrigin-RevId: 0c568d3335
|
4 years ago |
Megvii Engine Team
|
aa953c3bd6
|
fix(mge/module): fix missing import
GitOrigin-RevId: 7cdf6ac814
|
4 years ago |
Megvii Engine Team
|
5a01de7851
|
fix(mge): fix scalar transpose
GitOrigin-RevId: c2b9e025c7
|
4 years ago |
Megvii Engine Team
|
8536864351
|
chore(version): fix dev version to a large number
GitOrigin-RevId: 9567513c1f
|
4 years ago |
Megvii Engine Team
|
61c5c9cf74
|
chore(cmake): normlize some cmake message level
GitOrigin-RevId: 4550cbfaa6
|
4 years ago |
Megvii Engine Team
|
2e87420865
|
fix(module): fix docs in normalization
GitOrigin-RevId: 326234c35a
|
4 years ago |
Megvii Engine Team
|
638ab52fdc
|
feat(mge/imperative): simulates scalar
GitOrigin-RevId: e81630e256
|
4 years ago |
Megvii Engine Team
|
7167fdbd49
|
feat(mge/module): add normalization module includes group_norm, instance_norm and layer_norm
GitOrigin-RevId: 9f253e3250
|
4 years ago |
Megvii Engine Team
|
94dba16ff4
|
perf(mge/imperative): misc optimizations
GitOrigin-RevId: bbe7a10b00
|
4 years ago |
Megvii Engine Team
|
442058aead
|
feat(mge/quantize): add batch matmul activation module for inference
GitOrigin-RevId: f6dbd1f4c0
|
4 years ago |
Megvii Engine Team
|
81e0dae4ac
|
docs(mge): update GradManger docs
GitOrigin-RevId: ccffb830c4
|
4 years ago |
Megvii Engine Team
|
6667100638
|
feat(mge): use weakref for GradManger.attach
GitOrigin-RevId: 6df336c3c1
|
4 years ago |
Megvii Engine Team
|
75ca5bfe65
|
feat(mge): remove GradManager.detach until it is ready
GitOrigin-RevId: 1302b25b2a
|
4 years ago |
Megvii Engine Team
|
544b798374
|
revert: feat(mge/grad_manager): add `clear_grad` method for GradManager
This reverts commit aa9540e090 .
GitOrigin-RevId: 2d15e503de
|
4 years ago |
Megvii Engine Team
|
cc85047bf0
|
fix(mge/trace): fix sublinear in trace
GitOrigin-RevId: 356dcd9523
|
4 years ago |
Megvii Engine Team
|
b9918c329d
|
feat(mge/distributed): support distributed key-value store
GitOrigin-RevId: b4abe80014
|
4 years ago |
Megvii Engine Team
|
ab82c8da79
|
feat(mge/device): add python binding for get_mem_status_bytes
GitOrigin-RevId: aa0d3264d6
|
4 years ago |
Megvii Engine Team
|
ab6328c563
|
feat(imperative): port persistent cache
GitOrigin-RevId: 8ca24a37cc
|
4 years ago |
Megvii Engine Team
|
13272eaa63
|
fix(mge/trace): fix random op in symbolic trace
GitOrigin-RevId: 9a851cd177
|
4 years ago |
Megvii Engine Team
|
1fed59293b
|
perf(mge/optimizer): close conver_inputs for optimizer step
GitOrigin-RevId: c710530d93
|
4 years ago |
Megvii Engine Team
|
495472954d
|
fix(trace): link io-op to avoid deadlock
GitOrigin-RevId: 872cb6b715
|
4 years ago |
Megvii Engine Team
|
33e8879af4
|
feat(mge/quantization): support distributed qat
GitOrigin-RevId: c915c843b8
|
4 years ago |
Megvii Engine Team
|
8e11204ab6
|
perf(mge/module): optimize conv_bn qat module to improve performance
GitOrigin-RevId: 9415b83d9c
|
4 years ago |
Megvii Engine Team
|
82b0f67770
|
fix(mge/core): fix dtype promotion issue for quantized dtype
GitOrigin-RevId: 9d09c8fa6f
|
4 years ago |
Megvii Engine Team
|
8118a5946d
|
fix(mge/utils): fix get_oprs_seq of cgtools
GitOrigin-RevId: 366a56f4d5
|
4 years ago |
Megvii Engine Team
|
ae8c3c8163
|
feat(mge/functional): add python wrapper for fake quant opr
GitOrigin-RevId: 5f205198be
|
4 years ago |
Megvii Engine Team
|
2e53077915
|
fix(mge/trace): use xpux device when dump
GitOrigin-RevId: f37285f70e
|
4 years ago |
Megvii Engine Team
|
9cb3c07cd4
|
feat(mge/functional): add elemwise mode support string input
GitOrigin-RevId: 57be5cec7b
|
4 years ago |
Megvii Engine Team
|
8fad00a158
|
feat(mge/functional/nn): add conv1d padding
GitOrigin-RevId: 1bbfd36b96
|
4 years ago |
Megvii Engine Team
|
094601e834
|
feat(mge/distributed): allow remote grad by using grad manager
GitOrigin-RevId: a890c206a5
|
4 years ago |
Megvii Engine Team
|
02438ee635
|
refactor(mge/distributed): use thread.Threading to create Server
GitOrigin-RevId: fc994411bf
|
4 years ago |
Megvii Engine Team
|
a1efd4d56d
|
fix(mge/parampack): fix param pack when no param left
GitOrigin-RevId: b88b876064
|
4 years ago |
Megvii Engine Team
|
e0e1896450
|
docs(mge): formatting math in docstring
GitOrigin-RevId: e6447a2786
|
4 years ago |
Megvii Engine Team
|
28dbadf729
|
feat(imperative): add remap opr
GitOrigin-RevId: 1fa7e18d4b
|
4 years ago |
Megvii Engine Team
|
1e0fb127d1
|
perf(mge/functional): reduce matmul python overhead
GitOrigin-RevId: 738d0da10e
|
4 years ago |
Megvii Engine Team
|
7cd846a570
|
fix(mge/batch_norm): fix batch_norm check when trace(symbolic=True)
GitOrigin-RevId: 2032eb5f7d
|
4 years ago |
Megvii Engine Team
|
c89209450d
|
ci(mge): remove blank line for windows doctest
GitOrigin-RevId: 09cf6518e8
|
4 years ago |
Megvii Engine Team
|
e082e27780
|
refactor(mge/data): Refactor megeninge.data.dataset
GitOrigin-RevId: 1d9c61ce70
|
4 years ago |
Megvii Engine Team
|
f04e0d777e
|
feat(mge/data): dpflow dataset, stream sampler and loader
GitOrigin-RevId: cbb4510a13
|
4 years ago |