Megvii Engine Team
df976782fa
feat(mge/functional): add matinv
GitOrigin-RevId: d4fa8a8277
4 years ago
Megvii Engine Team
169fa53d54
fix(mgb): fix execution_policy set of matmul
GitOrigin-RevId: 90f539b0be
4 years ago
Megvii Engine Team
abe3c165ba
feat(mge): tensorrt runtime opr
GitOrigin-RevId: 2fdd00adcb
4 years ago
Megvii Engine Team
1b1ad56a82
fix(mgb/gopt): fix warp fusion opt pass
GitOrigin-RevId: a40bbcd719
4 years ago
Megvii Engine Team
99b176236b
fix(mgb/opr): fix Reduce static value inference
GitOrigin-RevId: 5e5c56064c
4 years ago
Megvii Engine Team
0f739c1118
fix(mgb/gopt): fix NormalizeArithChainPass to process sub-div chains
GitOrigin-RevId: d71debbfde
4 years ago
Megvii Engine Team
45e206020a
feat(mgb/tensor): add proxy_to_comp_node for HostTensorND
GitOrigin-RevId: 8f03dbcbd1
4 years ago
Megvii Engine Team
5db17907b4
chore(megbrain): fix megskull cnn/img opr compatiblity
for add convolutionv2 param
GitOrigin-RevId: 9d0cf55ef4
4 years ago
Megvii Engine Team
935fe7819e
chore(megbrain): format sereg.h code
GitOrigin-RevId: 0fe1cf6a8b
4 years ago
Megvii Engine Team
55042195d4
chore(winograd): add Convolutionv2 param
GitOrigin-RevId: 1a9e2ea340
4 years ago
Megvii Engine Team
fe6af7cb8d
fix(mgb): fix fastrun log
GitOrigin-RevId: 0a6cd47e49
4 years ago
Megvii Engine Team
f79ed946e3
fix(mgb/cambricon): support uint8 input for cambricon runtime opr
GitOrigin-RevId: 7c35662ddd
4 years ago
Megvii Engine Team
e9a7be46b1
feat(mgb): adapt to imperative runtime
GitOrigin-RevId: 3bccc17b62
4 years ago
Megvii Engine Team
5063a2063b
feat(mgb): add fastrun support for matmul/batchedmatrixmul
GitOrigin-RevId: a48ea9bff6
4 years ago
Megvii Engine Team
27503620a5
fix(fastrun): release free blocks before profiling
GitOrigin-RevId: b0d9887e9a
4 years ago
Megvii Engine Team
62637fc46e
refactor(mgb/utils): allow user set different spin time for ASyncSCQueue
GitOrigin-RevId: 1c6c4a7a16
4 years ago
Megvii Engine Team
d07cfdcbe9
refactor(mgb): move convolution mixin to search_policy
GitOrigin-RevId: 81e32da034
4 years ago
Megvii Engine Team
a57317f4e1
refactor(mgb): move profile cache out of mgb opr and update CACHE_KEY_VERSION
GitOrigin-RevId: db433164d2
4 years ago
Megvii Engine Team
ebe8689267
feat(sdk/load_and_run): add flags --io-dump-stdout and --io-dump-stderr
GitOrigin-RevId: 64d572bdfc
4 years ago
Megvii Engine Team
a240d558f1
fix(mge/dnn): fix rng and topk oom in distributed training
GitOrigin-RevId: 9841d1219e
4 years ago
Megvii Engine Team
c3c1e46df7
chore(mgblar): delete winograd transform switch
GitOrigin-RevId: aa4051541d
4 years ago
Megvii Engine Team
3975a54aa4
fix(mge/sublinear): fix divide by zero in sublinear
GitOrigin-RevId: d2bd113809
4 years ago
Megvii Engine Team
8f7f52ae4d
feat(jit): add memfwd in jit executor opr
GitOrigin-RevId: b58860bbe8
4 years ago
Megvii Engine Team
dfb2b2ce49
fix(dnn): change pooling window size smaller than padding constraint to log_error
GitOrigin-RevId: c3cda68f6d
4 years ago
Megvii Engine Team
c49427d15a
feat(imperative): add inplace add_update option in optimizer
GitOrigin-RevId: b8feb49321
4 years ago
Megvii Engine Team
aeb7980b29
perf(mgb): outputs of the same opr and same compnode share the same callbackcaller
GitOrigin-RevId: 59b8e3bcbe
4 years ago
Megvii Engine Team
7f3f9a94ae
feat(mgb/core): add shape hint for graph optimization
GitOrigin-RevId: eaad25a7ef
4 years ago
Megvii Engine Team
a85531dd0f
feat(mgb/opr): add tqt opr
GitOrigin-RevId: 49c62cd532
4 years ago
Megvii Engine Team
a5609f3bbf
fix(cambricon): fix cross cn copy for cambricon
GitOrigin-RevId: 21942a82a3
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
14d8b709e1
perf(mge/imperative): add mini graph to partially replace proxy graph
GitOrigin-RevId: 73e2529ba5
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
4e9be159f7
feat(mgb/gopt): add opt pass for fusing convolution and reformat
GitOrigin-RevId: d0c5deace2
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
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
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
fc0fcd2f7f
chore(winograd): remove winograd transform code
GitOrigin-RevId: 78c3cfceae
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
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
3bf73ff16f
feat(dnn): add cuda preprocess fusion
GitOrigin-RevId: d789c99e59
4 years ago
Megvii Engine Team
5f171298aa
feat(mgb/gopt): add AxisAddRemove opr support for cd4 opt pass
GitOrigin-RevId: 85218ee0c4
4 years ago
Megvii Engine Team
dbb64b46d5
feat(debug/android): opt android backtrace
GitOrigin-RevId: d8cef5f7c1
4 years ago