Megvii Engine Team
|
89b6dbc74c
|
fix(mge/imperative): import subpackages
GitOrigin-RevId: 342f6c87cc
|
4 years ago |
Megvii Engine Team
|
a85531dd0f
|
feat(mgb/opr): add tqt opr
GitOrigin-RevId: 49c62cd532
|
4 years ago |
Megvii Engine Team
|
44742e32e9
|
fix(mge/api): check input dim of dot and mark output as scalar
GitOrigin-RevId: a3ba7e099c
|
4 years ago |
Megvii Engine Team
|
05c739b846
|
refactor(mge/data): rename `MapDataset` to `Dataset`
GitOrigin-RevId: 6262561355
|
4 years ago |
Megvii Engine Team
|
a892e5d0e4
|
perf(mge): add more specialized grad rules
GitOrigin-RevId: f88809a6d7
|
4 years ago |
Megvii Engine Team
|
75129cf071
|
chore(mge): clean up before merge to dev
* remove dead test
* clean some codes
* fix test_fake_quant due to change of implementation
GitOrigin-RevId: f030a9966d
|
4 years ago |
Megvii Engine Team
|
dd9f54cdfa
|
refactor(mge): migrate to new core implementation
* remove dispatcher/interpreter python wrapper
* rename tensor_wrapper to array_method
GitOrigin-RevId: b8a402c2be
|
4 years ago |
Megvii Engine Team
|
b9762d714c
|
fix(mge): make parampack run with tensor symbolic shape
GitOrigin-RevId: 6fc313785d
|
4 years ago |
Megvii Engine Team
|
4d75f691a0
|
feat(mge): restore remote send/recv
GitOrigin-RevId: 8b78fd5591
|
4 years ago |
Megvii Engine Team
|
9c92701f63
|
feat(mge): support python -m megengine.distributed.server
GitOrigin-RevId: f1e5c8e3cf
|
4 years ago |
Megvii Engine Team
|
556e022215
|
refactor(mge/trace): remove apply on varnode
GitOrigin-RevId: 0185244c1f
|
4 years ago |
Megvii Engine Team
|
243a05b410
|
fix(mge): fix cpp trace function release
GitOrigin-RevId: 73f9642821
|
4 years ago |
Megvii Engine Team
|
60c7d62a4c
|
refactor(imperative): remove multidispatch, raw_tensor, register
GitOrigin-RevId: ca5a6ed8eb
|
4 years ago |
Megvii Engine Team
|
b5e46ae92f
|
feat(mge): restore Function
GitOrigin-RevId: dd455238ba
|
4 years ago |
Megvii Engine Team
|
ae47fd4e2a
|
fix(mge): fix none return value for attrs, add test_correctness
GitOrigin-RevId: 1bb96373f4
|
4 years ago |
Megvii Engine Team
|
87f4b46e3f
|
perf(mge/imperative): move convert_inputs from python to C++
GitOrigin-RevId: baef3d348c
|
4 years ago |
Megvii Engine Team
|
b310f2615b
|
feat(mge/imperative): implement trace and dump under new core implementation
GitOrigin-RevId: 4edc38eaf2
|
4 years ago |
Megvii Engine Team
|
c294b9d18b
|
refactor(mge/tensor): remove old implementation
remove core.tensor, raw_tensor,TensorWrapper
avoid create tensor with zero-stride numpy ndarray
GitOrigin-RevId: 4fe5c4c5ba
|
4 years ago |
Megvii Engine Team
|
15e8e7df76
|
refactor(mge/imperative): migrate to new core implementation
* swap/drop
* config/set_async_level
* _dev_tensor
* sync
GitOrigin-RevId: 850fb98852
|
4 years ago |
Megvii Engine Team
|
34c705fcbf
|
refactor(mge/imperative): move detach into C++
GitOrigin-RevId: 8c0d86cbbf
|
4 years ago |
Megvii Engine Team
|
147cef52a0
|
refactor(mge/imperative): implement new tensor system
GitOrigin-RevId: 2dd4e460ac
|
4 years ago |
Megvii Engine Team
|
577366a203
|
feat(mge/imperative): basic impl of python c extension for opdef
fix(imperative): fix refcount management on cpython opdef
refactor(mge/imperative): fix compilation for python3.6
GitOrigin-RevId: 332a516895
|
4 years ago |
Megvii Engine Team
|
9d928e7f83
|
refactor(mge/distributed): sync interpreter for distribtued launcher
GitOrigin-RevId: 8a88c272a1
|
4 years ago |
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 |