Megvii Engine Team
|
13272eaa63
|
fix(mge/trace): fix random op in symbolic trace
GitOrigin-RevId: 9a851cd177
|
4 years ago |
Megvii Engine Team
|
1fed59293b
|
perf(mge/optimizer): close conver_inputs for optimizer step
GitOrigin-RevId: c710530d93
|
4 years ago |
Megvii Engine Team
|
1f75c7ade4
|
ci(midout): fix midout and reopen midout test
we just test function, do not check size
GitOrigin-RevId: dce5387e83
|
4 years ago |
Megvii Engine Team
|
495472954d
|
fix(trace): link io-op to avoid deadlock
GitOrigin-RevId: 872cb6b715
|
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
|
798f7b3ef6
|
feat(imperative): add virtual_deps op
GitOrigin-RevId: 5845520989
|
4 years ago |
Megvii Engine Team
|
1e71e0afe0
|
refactor(dnn): refactor deconv algo
GitOrigin-RevId: 422be792eb
|
4 years ago |
Megvii Engine Team
|
89ad33aeb3
|
feat(dnn/cuda): support weight preprocessing for cutlass algorithms
GitOrigin-RevId: 7b77579acd
|
4 years ago |
Megvii Engine Team
|
33e8879af4
|
feat(mge/quantization): support distributed qat
GitOrigin-RevId: c915c843b8
|
4 years ago |
Megvii Engine Team
|
8e11204ab6
|
perf(mge/module): optimize conv_bn qat module to improve performance
GitOrigin-RevId: 9415b83d9c
|
4 years ago |
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 |