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
Megvii Engine Team
e8bf5bc002
fix(mge): ensure contiguous when passing tensor from graph rt to imperative rt
GitOrigin-RevId: e4d944343d
4 years ago
Megvii Engine Team
3af1056377
feat(mge/grad): attach grad immediately
GitOrigin-RevId: e3a168c03a
4 years ago
Megvii Engine Team
dc3c17ba13
chore(cmake): fix a cmake typo
GitOrigin-RevId: e2b59f2f15
5 years ago
Megvii Engine Team
44b27f0d6e
build(3516): fix some cpu flags build failed and fix 3516 ycm
GitOrigin-RevId: a0c73fa18a
4 years ago
Megvii Engine Team
cab46f2a00
fix(dnn): fix relayout format when group conv group=1
GitOrigin-RevId: f2b53be77f
4 years ago
Megvii Engine Team
8b154fd7a5
fix(license): correct and format the content of license to make it be detected by GitHub
GitOrigin-RevId: 00e4972023
4 years ago
Megvii Engine Team
b13c78a532
fix(ios): fix cmake ios cross build
GitOrigin-RevId: 64e74594ce
4 years ago
Megvii Engine Team
cfc430b8c7
fix(mbg/opr): fix ModifySubtensorImplHelper infer shape depend input value for valid check
GitOrigin-RevId: 5ad6b9f35c
4 years ago
Megvii Engine Team
8764a6c8ff
feat(dnn/cuda): add volta dp4a int8 sass kernel
GitOrigin-RevId: 9fefd39678
4 years ago
Megvii Engine Team
e296a6842a
fix(build/doc): fix build md
GitOrigin-RevId: 7cb159666d
4 years ago
Megvii Engine Team
3635af6274
style(atlas): add comment for async d2d
GitOrigin-RevId: 606a56ac4e
4 years ago
Megvii Engine Team
d68d4d1d99
perf(atlas): use async d2d
GitOrigin-RevId: 55914631cb
4 years ago
Megvii Engine Team
d8ac6c70a1
fix(quantize): fix quantize calibration dtype issue
GitOrigin-RevId: 667c99f469
4 years ago
Megvii Engine Team
4afa4b72c3
fix(mgb/oppass): fix fuse conv bias pass when bias is full bias and bias
is const provider
GitOrigin-RevId: 3285022f33
4 years ago
Megvii Engine Team
50d64bd333
fix(pytest/windows/impertive): open impertive pytest on windows
GitOrigin-RevId: 2e8f15607e
4 years ago
Megvii Engine Team
dfc3935d06
fix(pytest/windows/impertive): fix windows wrong test skip
GitOrigin-RevId: ca0f04cfe7
4 years ago
Megvii Engine Team
786399c5c9
fix(pytest/windows/impertive): fix impertive pytest failed on windows
GitOrigin-RevId: 02f4c0a0be
4 years ago