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
Megvii Engine Team
ab9f44f15c
feat(mge/quantization): add support for easyquant
GitOrigin-RevId: 060d908349
4 years ago
Megvii Engine Team
fc0fcd2f7f
chore(winograd): remove winograd transform code
GitOrigin-RevId: 78c3cfceae
4 years ago
Megvii Engine Team
d1adc9a22f
fix(dnn): fix opencl algo search
GitOrigin-RevId: 25997d0ef1
4 years ago
Megvii Engine Team
368c18607f
fix(mgb/jit): find cuda include path correctly
GitOrigin-RevId: 5ced4206a7
4 years ago
Megvii Engine Team
b04e0466bb
feat(megbrain): add alias name to model serialization
GitOrigin-RevId: d47e650473
4 years ago
Megvii Engine Team
cf53d9e0f8
fix(mgb/tensor): do tensor overlap check only when d2d and h2h
GitOrigin-RevId: 9125936a35
4 years ago
Megvii Engine Team
7e2b2dbffc
fix(dnn/test): delete large size in ARM_COMMON.FP32_GEVM
GitOrigin-RevId: 581ef43816
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
0398a7867f
fix(build/windows/cuda/llvm): fix windows bazel build with cuda
* Adapt to the new version llvm/clang-11
* fix windows bazel build with cuda
* add windows bazel build cuda ci
* opt windows bazel ci scripts
GitOrigin-RevId: 6ea7c66585
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
3bf73ff16f
feat(dnn): add cuda preprocess fusion
GitOrigin-RevId: d789c99e59
4 years ago
Megvii Engine Team
86cf7490ec
feat(dnn/aarch64): add quantizeds4 matmul int4x4x16_k8x8x8
GitOrigin-RevId: 7812900244
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
142f31a875
perf(dnn/cuda): change conv_bias heu, prefer dnn chanwise impl, dislike dnn batch gemm conv1x1
GitOrigin-RevId: 323bf6073a
4 years ago
Megvii Engine Team
f214e14695
refactor(mgb/cuda): use single implementation of get_device_prop from utils
GitOrigin-RevId: 5cc95472b9
4 years ago
Megvii Engine Team
54e79dd1d9
perf(mgb/cuda): do not call cudaGetDeviceProperties to avoid io traffic
GitOrigin-RevId: 6aa35928c8
4 years ago
Megvii Engine Team
5f171298aa
feat(mgb/gopt): add AxisAddRemove opr support for cd4 opt pass
GitOrigin-RevId: 85218ee0c4
4 years ago