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
|
7f48625fd7
|
chore(third_party): add range-v3 as third_party dependency
GitOrigin-RevId: cdb5a9ef3b
|
4 years ago |
Megvii Engine Team
|
d168cea4a7
|
feat(opr): add param(axis) for GetVarShape
feat(mge/imperative): GetVarShape support negative axis
GitOrigin-RevId: 30ce0758e6
|
4 years ago |
Megvii Engine Team
|
e9c036ccba
|
fix(mge/imperative): fix op specializations and tuple hash
GitOrigin-RevId: e3df93fd5b
|
4 years ago |
Megvii Engine Team
|
b5ec83c53f
|
test(mge/autodiff): validate grad is set to None when no dependency
GitOrigin-RevId: 01ad51a222
|
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
|
4cfedc164a
|
build(imperative): clean develop before build imperative and mgb
GitOrigin-RevId: 4b56e148d8
|
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
|
9de1ea6a48
|
perf(imperative): add apply_on_physical_tensor for Elemwise
GitOrigin-RevId: 27087d90e4
|
4 years ago |
Megvii Engine Team
|
a78c11099b
|
fix(imperative): add param(axis) for GetVarShape
GitOrigin-RevId: 0b8f821929
|
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
|
bff0fc6172
|
fix(mge/interpreter): fix outputs check on async level0
GitOrigin-RevId: 46c6a2ce70
|
4 years ago |
Megvii Engine Team
|
93f4977c78
|
feat(mge/imperative): add thread name
GitOrigin-RevId: 61eece55b8
|
4 years ago |
Megvii Engine Team
|
57546b4c3d
|
test(mge/distributed): fix test skip condition error
GitOrigin-RevId: f540c53302
|
4 years ago |
Megvii Engine Team
|
783a612643
|
feat(debug/macos/windows): imp macos/windows backtrace, fix mem issue
GitOrigin-RevId: 9bed738bb5
|
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
|
9f139562b4
|
test(mge): fix requires-test.txt for doctest
GitOrigin-RevId: 3d92febedd
|
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
|
176268d20c
|
test(mge): disable gc in leak check
GitOrigin-RevId: b650a46e02
|
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 |