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
Megvii Engine Team
3e00e3f697
feat(debug/linux): opt linux backtrace
GitOrigin-RevId: bdf554cd3c
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
e92670e820
fix(mgb/atlas): when batchsize more than model max batchsize
GitOrigin-RevId: 63fe79eaa9
4 years ago
Megvii Engine Team
147dbf8a0c
fix(test): fix a race condition in TestCudaMemAlloc
GitOrigin-RevId: 99468a7ae3
4 years ago
Megvii Engine Team
7066ad5ba6
feat(dnn): add uint16 support
GitOrigin-RevId: f4c4b1c7b9
4 years ago
Megvii Engine Team
a1877ee0fa
refactor(dnn): refactor algo interface, use algoinfo instead of global algorithm
GitOrigin-RevId: 479718ac75
4 years ago
Megvii Engine Team
cb59c27835
feat(mlir/ir): add more op definitions
GitOrigin-RevId: 1e1285ef41
4 years ago
Megvii Engine Team
9ec8d375f1
feat(externcopr): add config extern c opr dynamic param
GitOrigin-RevId: 0fa3a534af
4 years ago
Megvii Engine Team
88e918e261
feat(mgb/jit): add scf.ForOp in MgbToGpuLoweringPass
GitOrigin-RevId: 3cdae27c37
4 years ago
Megvii Engine Team
6f5d0febf1
perf(dnn/cuda): enhance performance for pooling forward
GitOrigin-RevId: 55fb2a9b25
4 years ago
Megvii Engine Team
f7731bd437
fix(mgb/jit): fix a pointer bug in mlir executable_cuda
GitOrigin-RevId: 3ec79b7602
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
f470df4f1a
fix(mgb/opr): fix convbias with no bias when weight preprocess
GitOrigin-RevId: 8837ad539c
4 years ago
Megvii Engine Team
6b9ac894d3
fix(mgb/topk): fix topk grad
GitOrigin-RevId: a49154ff06
4 years ago
Megvii Engine Team
61c5c9cf74
chore(cmake): normlize some cmake message level
GitOrigin-RevId: 4550cbfaa6
4 years ago
Megvii Engine Team
5d0f8da46a
feat(mgb/jit): add Dimshuffle and lowering passes in jit mlir backend
GitOrigin-RevId: ce6f4ea42a
4 years ago
Megvii Engine Team
0007b9e09b
build(third_party): update llvm-project
GitOrigin-RevId: 26bb606606
4 years ago
Megvii Engine Team
404ef808fa
feat(mgb/jit): adapt jit mlir backend to new mgb dialect and add typecvt
GitOrigin-RevId: bd1b80c84f
4 years ago
Megvii Engine Team
60c6d59fc9
feat(mbg/core): support bias preprocess in conv_bias
GitOrigin-RevId: d2e1e14d41
4 years ago
Megvii Engine Team
064c774f05
feat(imperative): impl hashable for SendRecv and add virtual input for Recv
GitOrigin-RevId: 5e8c27ac81
4 years ago
Megvii Engine Team
b60cc8ca3e
feat(mgb/opr): add megbrain fake quant opr
GitOrigin-RevId: a858bce939
4 years ago
Megvii Engine Team
4f9948d00c
chore(mbg/core): add MGB_WORKER_SHORT_SPIN env variable to set short spin
GitOrigin-RevId: c757dc36d3
4 years ago
Megvii Engine Team
4f527af4ae
chore(mbg/core): fix the repeat code
GitOrigin-RevId: 3d4c91a2c9
4 years ago
Megvii Engine Team
fea5cf6279
fix(sdk/loader): debug sstar loader
GitOrigin-RevId: a5847ad460
4 years ago
Megvii Engine Team
d793c87cdd
refactor(mlir/dialect): redefine mgb dialect
GitOrigin-RevId: ac06534746
4 years ago
Megvii Engine Team
005ead5aea
docs(cpu/comp_node): note cpu_default()
GitOrigin-RevId: 3e3dbced65
4 years ago