Megvii Engine Team
a78c11099b
fix(imperative): add param(axis) for GetVarShape
GitOrigin-RevId: 0b8f821929
5 years ago
Megvii Engine Team
fc0fcd2f7f
chore(winograd): remove winograd transform code
GitOrigin-RevId: 78c3cfceae
5 years ago
Megvii Engine Team
368c18607f
fix(mgb/jit): find cuda include path correctly
GitOrigin-RevId: 5ced4206a7
5 years ago
Megvii Engine Team
b04e0466bb
feat(megbrain): add alias name to model serialization
GitOrigin-RevId: d47e650473
5 years ago
Megvii Engine Team
cf53d9e0f8
fix(mgb/tensor): do tensor overlap check only when d2d and h2h
GitOrigin-RevId: 9125936a35
5 years ago
Megvii Engine Team
69e3e32240
feat(imperative): auto generated opdef header and python binding
GitOrigin-RevId: d2f22ad5fe
5 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
5 years ago
Megvii Engine Team
3bf73ff16f
feat(dnn): add cuda preprocess fusion
GitOrigin-RevId: d789c99e59
5 years ago
Megvii Engine Team
5f171298aa
feat(mgb/gopt): add AxisAddRemove opr support for cd4 opt pass
GitOrigin-RevId: 85218ee0c4
5 years ago
Megvii Engine Team
dbb64b46d5
feat(debug/android): opt android backtrace
GitOrigin-RevId: d8cef5f7c1
5 years ago
Megvii Engine Team
3e00e3f697
feat(debug/linux): opt linux backtrace
GitOrigin-RevId: bdf554cd3c
5 years ago
Megvii Engine Team
783a612643
feat(debug/macos/windows): imp macos/windows backtrace, fix mem issue
GitOrigin-RevId: 9bed738bb5
5 years ago
Megvii Engine Team
e92670e820
fix(mgb/atlas): when batchsize more than model max batchsize
GitOrigin-RevId: 63fe79eaa9
5 years ago
Megvii Engine Team
147dbf8a0c
fix(test): fix a race condition in TestCudaMemAlloc
GitOrigin-RevId: 99468a7ae3
5 years ago
Megvii Engine Team
7066ad5ba6
feat(dnn): add uint16 support
GitOrigin-RevId: f4c4b1c7b9
5 years ago
Megvii Engine Team
a1877ee0fa
refactor(dnn): refactor algo interface, use algoinfo instead of global algorithm
GitOrigin-RevId: 479718ac75
5 years ago
Megvii Engine Team
cb59c27835
feat(mlir/ir): add more op definitions
GitOrigin-RevId: 1e1285ef41
5 years ago
Megvii Engine Team
9ec8d375f1
feat(externcopr): add config extern c opr dynamic param
GitOrigin-RevId: 0fa3a534af
5 years ago
Megvii Engine Team
88e918e261
feat(mgb/jit): add scf.ForOp in MgbToGpuLoweringPass
GitOrigin-RevId: 3cdae27c37
5 years ago
Megvii Engine Team
6f5d0febf1
perf(dnn/cuda): enhance performance for pooling forward
GitOrigin-RevId: 55fb2a9b25
5 years ago
Megvii Engine Team
f7731bd437
fix(mgb/jit): fix a pointer bug in mlir executable_cuda
GitOrigin-RevId: 3ec79b7602
5 years ago
Megvii Engine Team
2ad8c5e1e9
fix(mge/io_remote): fix remote send/recv gradient at trace
GitOrigin-RevId: 7886efd0c1
5 years ago
Megvii Engine Team
f470df4f1a
fix(mgb/opr): fix convbias with no bias when weight preprocess
GitOrigin-RevId: 8837ad539c
5 years ago
Megvii Engine Team
6b9ac894d3
fix(mgb/topk): fix topk grad
GitOrigin-RevId: a49154ff06
5 years ago
Megvii Engine Team
61c5c9cf74
chore(cmake): normlize some cmake message level
GitOrigin-RevId: 4550cbfaa6
5 years ago
Megvii Engine Team
5d0f8da46a
feat(mgb/jit): add Dimshuffle and lowering passes in jit mlir backend
GitOrigin-RevId: ce6f4ea42a
5 years ago
Megvii Engine Team
0007b9e09b
build(third_party): update llvm-project
GitOrigin-RevId: 26bb606606
5 years ago
Megvii Engine Team
404ef808fa
feat(mgb/jit): adapt jit mlir backend to new mgb dialect and add typecvt
GitOrigin-RevId: bd1b80c84f
5 years ago
Megvii Engine Team
60c6d59fc9
feat(mbg/core): support bias preprocess in conv_bias
GitOrigin-RevId: d2e1e14d41
5 years ago
Megvii Engine Team
064c774f05
feat(imperative): impl hashable for SendRecv and add virtual input for Recv
GitOrigin-RevId: 5e8c27ac81
5 years ago
Megvii Engine Team
b60cc8ca3e
feat(mgb/opr): add megbrain fake quant opr
GitOrigin-RevId: a858bce939
5 years ago
Megvii Engine Team
4f9948d00c
chore(mbg/core): add MGB_WORKER_SHORT_SPIN env variable to set short spin
GitOrigin-RevId: c757dc36d3
5 years ago
Megvii Engine Team
4f527af4ae
chore(mbg/core): fix the repeat code
GitOrigin-RevId: 3d4c91a2c9
5 years ago
Megvii Engine Team
fea5cf6279
fix(sdk/loader): debug sstar loader
GitOrigin-RevId: a5847ad460
5 years ago
Megvii Engine Team
d793c87cdd
refactor(mlir/dialect): redefine mgb dialect
GitOrigin-RevId: ac06534746
5 years ago
Megvii Engine Team
005ead5aea
docs(cpu/comp_node): note cpu_default()
GitOrigin-RevId: 3e3dbced65
5 years ago
Megvii Engine Team
9415ba5890
feat(src/core): free weight preprocessed weight
GitOrigin-RevId: 5f91acb909
5 years ago
Megvii Engine Team
7cd71c3102
fix(mgb/gopt): fix cd4 elewise transform
GitOrigin-RevId: 027d5e53e4
5 years ago
Megvii Engine Team
280861aebb
feat(atlas): add acl.json support, use for profile
GitOrigin-RevId: c123806fef
5 years ago
Megvii Engine Team
9682db9820
feat(mgb): add jit mlir elemwise broadcast
GitOrigin-RevId: 89d5e2f91e
5 years ago
Megvii Engine Team
a3550f91ca
test(rocm): Fix rocm testcase
GitOrigin-RevId: d1a17d3edd
5 years ago
Megvii Engine Team
1e90c45793
refactor(mgb/opr): move fastrun out of conv opr
GitOrigin-RevId: d5ef5356f6
5 years ago
Megvii Engine Team
2a3f4d099a
refactor(dnn/arm): refactor CPU heuristic algo selection
GitOrigin-RevId: 60d2646bb3
5 years ago
Megvii Engine Team
5cfe73657b
fix(mgb/core): ensure all VarNodes would be handled in init_ready_event
GitOrigin-RevId: 0b6cb24021
5 years ago
Megvii Engine Team
2d0c96906a
fix(opr): add mask check for dct
GitOrigin-RevId: a452092cc7
5 years ago
Megvii Engine Team
45032220a4
feat(opr): add dct opr
GitOrigin-RevId: 3e2cc58cfe
5 years ago
Megvii Engine Team
170897f2e1
feat(mgb/jit): add more elemwise modes for mlir backend
GitOrigin-RevId: 5883c68804
5 years ago
Megvii Engine Team
ca717806af
fix(mgb): fix wrong use of macro MGB_ENABLE_GRAD
GitOrigin-RevId: e66dabee01
5 years ago
Megvii Engine Team
cab46f2a00
fix(dnn): fix relayout format when group conv group=1
GitOrigin-RevId: f2b53be77f
5 years ago
Megvii Engine Team
cfc430b8c7
fix(mbg/opr): fix ModifySubtensorImplHelper infer shape depend input value for valid check
GitOrigin-RevId: 5ad6b9f35c
5 years ago