Megvii Engine Team
|
d1fbec4fe2
|
feat(dnn/atlas): add atlas stub
GitOrigin-RevId: c63294378e
|
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
|
697f70c011
|
feat(mge/pytest): add more tests for specialized grad rules
GitOrigin-RevId: 509ef5a220
|
4 years ago |
Megvii Engine Team
|
cf3f58cb9b
|
fix(mge/autodiff): fix segfault when grad is nullptr
GitOrigin-RevId: 6139212bfd
|
4 years ago |
Megvii Engine Team
|
288c2e08b5
|
fix(mge/autodiff): fix expand_dims and grad rule fallback
GitOrigin-RevId: 4aae771222
|
4 years ago |
Megvii Engine Team
|
a5609f3bbf
|
fix(cambricon): fix cross cn copy for cambricon
GitOrigin-RevId: 21942a82a3
|
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
|
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 |