Megvii Engine Team
51fa530d2a
fix(mge/interpreter): add check for invalid tensor ptr
GitOrigin-RevId: e8edcd92a4
4 years ago
Megvii Engine Team
634de59061
feat(mge/imperative): add valid flag of `infer_output_attrs_fallible`
GitOrigin-RevId: b2b32774ee
4 years ago
Megvii Engine Team
50c4daac85
feat(mge/interpreter): add async_level mechanism for Interpreter
GitOrigin-RevId: 8615a23b75
4 years ago
Megvii Engine Team
82b0f67770
fix(mge/core): fix dtype promotion issue for quantized dtype
GitOrigin-RevId: 9d09c8fa6f
4 years ago
Megvii Engine Team
8118a5946d
fix(mge/utils): fix get_oprs_seq of cgtools
GitOrigin-RevId: 366a56f4d5
4 years ago
Megvii Engine Team
ae8c3c8163
feat(mge/functional): add python wrapper for fake quant opr
GitOrigin-RevId: 5f205198be
4 years ago
Megvii Engine Team
b60cc8ca3e
feat(mgb/opr): add megbrain fake quant opr
GitOrigin-RevId: a858bce939
4 years ago
Megvii Engine Team
c03249c059
feat(dnn/opr): add megdnn fake quant opr
GitOrigin-RevId: 5a04b6da2f
4 years ago
Megvii Engine Team
2e53077915
fix(mge/trace): use xpux device when dump
GitOrigin-RevId: f37285f70e
4 years ago
Megvii Engine Team
739f927c4c
feat(dnn/cuda): opt dp4a conv for small channel base on cutlass
GitOrigin-RevId: 2a74c35f27
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
1f8e40753f
fix(mkl): fix windows mkl LOG compute exception
GitOrigin-RevId: cd2ebaaec1
4 years ago
Megvii Engine Team
fea5cf6279
fix(sdk/loader): debug sstar loader
GitOrigin-RevId: a5847ad460
4 years ago
Megvii Engine Team
9cb3c07cd4
feat(mge/functional): add elemwise mode support string input
GitOrigin-RevId: 57be5cec7b
4 years ago
Megvii Engine Team
8fad00a158
feat(mge/functional/nn): add conv1d padding
GitOrigin-RevId: 1bbfd36b96
4 years ago
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