Megvii Engine Team
4aa277a203
refactor(dnn/cuda): misc
GitOrigin-RevId: 1f8f91a0cc
4 years ago
Megvii Engine Team
094601e834
feat(mge/distributed): allow remote grad by using grad manager
GitOrigin-RevId: a890c206a5
4 years ago
Megvii Engine Team
f7b2bdae1a
refactor(dnn): refactor algorithm type interface
GitOrigin-RevId: 843d885f82
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
Megvii Engine Team
9415ba5890
feat(src/core): free weight preprocessed weight
GitOrigin-RevId: 5f91acb909
4 years ago
Megvii Engine Team
7cd71c3102
fix(mgb/gopt): fix cd4 elewise transform
GitOrigin-RevId: 027d5e53e4
4 years ago
Megvii Engine Team
cae8c8a4a4
test(mge/parampack): add test for parampack
GitOrigin-RevId: f3e423dffc
4 years ago
Megvii Engine Team
02438ee635
refactor(mge/distributed): use thread.Threading to create Server
GitOrigin-RevId: fc994411bf
4 years ago
Megvii Engine Team
a1efd4d56d
fix(mge/parampack): fix param pack when no param left
GitOrigin-RevId: b88b876064
4 years ago
Megvii Engine Team
e0e1896450
docs(mge): formatting math in docstring
GitOrigin-RevId: e6447a2786
4 years ago
Megvii Engine Team
18ec5341f2
refactor(dnn): remove unused costmodel in cuda
GitOrigin-RevId: b15f0607b9
4 years ago
Megvii Engine Team
e39f938662
refactor(dnn): remove ProfileCache and matmul algo in x86
GitOrigin-RevId: 55a700d747
4 years ago
Megvii Engine Team
28dbadf729
feat(imperative): add remap opr
GitOrigin-RevId: 1fa7e18d4b
4 years ago
Megvii Engine Team
280861aebb
feat(atlas): add acl.json support, use for profile
GitOrigin-RevId: c123806fef
4 years ago
Megvii Engine Team
1e0fb127d1
perf(mge/functional): reduce matmul python overhead
GitOrigin-RevId: 738d0da10e
4 years ago
Megvii Engine Team
72619bb4fb
fix(whl/linux): fix mv failed if output have old whl
GitOrigin-RevId: c03ed56c61
4 years ago
Megvii Engine Team
9682db9820
feat(mgb): add jit mlir elemwise broadcast
GitOrigin-RevId: 89d5e2f91e
4 years ago
Megvii Engine Team
89303cd829
feat(megdnn/rocm): add bn for rocm backend
GitOrigin-RevId: 8bd49599b2
4 years ago
Megvii Engine Team
7cd846a570
fix(mge/batch_norm): fix batch_norm check when trace(symbolic=True)
GitOrigin-RevId: 2032eb5f7d
4 years ago
Megvii Engine Team
120e719e17
refactor(imperative): refactor OpTrait methods and registration
remove unused op methods and enable OpTrait registered from seperated files, also fix cond_take's infer_output_attrs
GitOrigin-RevId: 134c8215ce
4 years ago
Megvii Engine Team
aea829c9fa
feat(megdnn/rocm): add average inclusive mode for pooling
GitOrigin-RevId: 08162bd1fb
4 years ago
Megvii Engine Team
a3550f91ca
test(rocm): Fix rocm testcase
GitOrigin-RevId: d1a17d3edd
4 years ago
Megvii Engine Team
1e90c45793
refactor(mgb/opr): move fastrun out of conv opr
GitOrigin-RevId: d5ef5356f6
4 years ago
Megvii Engine Team
2fc77caf44
fix(whl/macos/windows/linux): fix macos/windows whl broken caused by
b2dbda96b, and add macos/windows whl import test, Integration Test will
test linux whl import, so we do not add linux import test at MR test
GitOrigin-RevId: 9aa88c2a43
4 years ago
Megvii Engine Team
c89209450d
ci(mge): remove blank line for windows doctest
GitOrigin-RevId: 09cf6518e8
4 years ago
Megvii Engine Team
e082e27780
refactor(mge/data): Refactor megeninge.data.dataset
GitOrigin-RevId: 1d9c61ce70
4 years ago
Megvii Engine Team
f04e0d777e
feat(mge/data): dpflow dataset, stream sampler and loader
GitOrigin-RevId: cbb4510a13
4 years ago
Megvii Engine Team
be511a56f9
fix(mge/imperative): fix tensor astype failed for quantized type
GitOrigin-RevId: 383458acbf
4 years ago
Megvii Engine Team
b309890c66
docs(mge): pytest for sphinx docstring
GitOrigin-RevId: 8bed12562a
4 years ago
Megvii Engine Team
1b56851715
fix(mge/trace): fix graph option in trace
GitOrigin-RevId: 7bec84f56d
4 years ago
Megvii Engine Team
f2f5f9acd6
fix(mgb/jit): remove MGE_USE_SYSTEM_LIB from mlir linking
GitOrigin-RevId: 57d7cbae36
4 years ago
Megvii Engine Team
1217801133
perf(mge): add opdef for broadcast
GitOrigin-RevId: 92f0af29eb
4 years ago
Megvii Engine Team
fccb2510c4
perf(mge): add opdef for elemwise and batchnorm
GitOrigin-RevId: d51fad9867
4 years ago
Megvii Engine Team
c008cf3757
refactor(mge): delete old dump_with_testcase_mge
GitOrigin-RevId: 3b58b4acd9
4 years ago
Megvii Engine Team
2a3f4d099a
refactor(dnn/arm): refactor CPU heuristic algo selection
GitOrigin-RevId: 60d2646bb3
4 years ago
Megvii Engine Team
981f487bf5
chore(third_party): update upstream of cutlass
4 years ago
Megvii Engine Team
5cfe73657b
fix(mgb/core): ensure all VarNodes would be handled in init_ready_event
GitOrigin-RevId: 0b6cb24021
4 years ago
Megvii Engine Team
2d0c96906a
fix(opr): add mask check for dct
GitOrigin-RevId: a452092cc7
4 years ago
Megvii Engine Team
9ffde21236
feat(python): add dct python opr
GitOrigin-RevId: 3814dec267
4 years ago
Megvii Engine Team
45032220a4
feat(opr): add dct opr
GitOrigin-RevId: 3e2cc58cfe
4 years ago
Megvii Engine Team
ba66e1d039
feat(dnn): add nchw_fp32 nchw44_qint8 cuda dct
GitOrigin-RevId: 581e31fc20
4 years ago
Megvii Engine Team
b327822994
feat(mge/grad_manager): add `clear_grad` method for GradManager
GitOrigin-RevId: aa9540e090
4 years ago
Megvii Engine Team
2627e1f7df
fix(mge/grad_manager): allow multiple calls of `release`
GitOrigin-RevId: 38ca4c78ff
4 years ago
Megvii Engine Team
67a543f350
build(mlir): remove unused llvm target to reduce compile time
GitOrigin-RevId: 6282d33efd
4 years ago
Megvii Engine Team
a9f98e9c66
refactor(meg/internal): move interal codes back to megbrain
GitOrigin-RevId: b2dbda96be
4 years ago
Megvii Engine Team
65c549137f
feat(load_and_run): feat --input parameter
GitOrigin-RevId: 2df8b6c164
4 years ago
Megvii Engine Team
8419a4fbc3
build(third_party): update llvm-project and adapt to mlir interface changes
4 years ago
Megvii Engine Team
170897f2e1
feat(mgb/jit): add more elemwise modes for mlir backend
GitOrigin-RevId: 5883c68804
4 years ago
Megvii Engine Team
ca717806af
fix(mgb): fix wrong use of macro MGB_ENABLE_GRAD
GitOrigin-RevId: e66dabee01
4 years ago