Megvii Engine Team
|
a892e5d0e4
|
perf(mge): add more specialized grad rules
GitOrigin-RevId: f88809a6d7
|
4 years ago |
Megvii Engine Team
|
e9e5f442a7
|
fix(mge): expand custom op before trace
GitOrigin-RevId: 725a5b87cb
|
4 years ago |
Megvii Engine Team
|
3faba54f28
|
fix(mge): fix segfault with Function returning unused grads
GitOrigin-RevId: 0cce845923
|
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
|
aba0acc797
|
fix(sdk): add AssertEqual opr, fix dump_with_testcase_mge
GitOrigin-RevId: 6f797570b6
|
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
|
9fb5581f1d
|
refactor(mge): add specialized grad rule support
GitOrigin-RevId: 141ff0a24f
|
4 years ago |
Megvii Engine Team
|
645fc6f0c8
|
test(mge/autodiff): add test for weird error
GitOrigin-RevId: e95109b845
|
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
|
dc2507453b
|
feat(mge): add python custom op
GitOrigin-RevId: 35da0bb301
|
4 years ago |
Megvii Engine Team
|
60c44b084c
|
refactor(mge): refactor to prepare for custom grad rules
GitOrigin-RevId: 4bd8850fdf
|
4 years ago |
Megvii Engine Team
|
61f65cd495
|
test(mge): fix megbrain_graph/cgtools test
GitOrigin-RevId: 33ac776b56
|
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
|
97d12b3e05
|
refactor(mge/imperative): fix compilation for Python 3.5
GitOrigin-RevId: edd49d0171
|
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
|
14d8b709e1
|
perf(mge/imperative): add mini graph to partially replace proxy graph
GitOrigin-RevId: 73e2529ba5
|
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
|
7f48625fd7
|
chore(third_party): add range-v3 as third_party dependency
GitOrigin-RevId: cdb5a9ef3b
|
4 years ago |
Megvii Engine Team
|
184e1311fa
|
fix(mgb/opr): move NVOF opr's shape inference to execute part
GitOrigin-RevId: 883c55e4a0
|
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
|
4e9be159f7
|
feat(mgb/gopt): add opt pass for fusing convolution and reformat
GitOrigin-RevId: d0c5deace2
|
4 years ago |
Megvii Engine Team
|
c3a4b2225d
|
feat(dnn/cuda): add cutlass impls for fused convolution reformat operation
GitOrigin-RevId: 02ef559c3f
|
4 years ago |
Megvii Engine Team
|
5f44203d7b
|
feat(dnn/cuda): add a cutlass impl for fusing convolution and dimshuffle
GitOrigin-RevId: 3fc6faef01
|
4 years ago |
Megvii Engine Team
|
61f917fb8e
|
feat(dnn/cuda): add impl for fusing warp perspective and dimshuffle
GitOrigin-RevId: 51e025973f
|
4 years ago |
Megvii Engine Team
|
15dd5e1a0f
|
fix(mgb/core): fix memory management release cambricon var issue
GitOrigin-RevId: abf881978c
|
4 years ago |
Megvii Engine Team
|
38ea5f1ba0
|
fix(mgb/jit): lower FuncOp to GPUFuncOp directly
GitOrigin-RevId: d7c9c0f548
|
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
|
d2910f7ef5
|
fix(mgb/jit): add bind_shape feature to MLIRCompiler
GitOrigin-RevId: bec6796fbf
|
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
|
469d0808f5
|
fix(asan/thread): fix asan issue
GitOrigin-RevId: 8e2b093fb1
|
4 years ago |
Megvii Engine Team
|
eb826422c4
|
fix(dnn): forbid pooling window size smaller than padding
GitOrigin-RevId: 9ad61c409d
|
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 |