Megvii Engine Team
|
042c7fd10a
|
feat(dnn): add elemwise modes
GitOrigin-RevId: edea1a48b0
|
3 years ago |
Megvii Engine Team
|
68d2710810
|
fix(build): remove build so many warning on windows
GitOrigin-RevId: 124b8975db
|
2 years ago |
Megvii Engine Team
|
dbd9483993
|
feat(dnn,src,imperative): add groupnorm op
GitOrigin-RevId: de3c3d10e5
|
2 years ago |
Megvii Engine Team
|
f444d4fe4d
|
feat(dnn,imperative): region restricted conv support groups=1 even if
depthwise
GitOrigin-RevId: 950d2f4889
|
2 years ago |
Megvii Engine Team
|
fdec82ece5
|
feat(dnn): add naive RegionRestrictedConv
GitOrigin-RevId: 5af93f7ccb
|
3 years ago |
Megvii Engine Team
|
a6a2646c10
|
feat(arm): add AlgoFP32Winograd F43, and add filter size into name of winograd-related algorithms
GitOrigin-RevId: 909503a90d
|
2 years ago |
Megvii Engine Team
|
f31e52d521
|
feat(mgb): warpperspective support multi src input
GitOrigin-RevId: 0887656864
|
2 years ago |
Megvii Engine Team
|
669816e291
|
feat(dnn): warpperspective support multi src input
GitOrigin-RevId: 8a4789852e
|
2 years ago |
Megvii Engine Team
|
8abc3ab8fc
|
fix(imperative): fix convolution in rocm
GitOrigin-RevId: 9e97099fd5
|
2 years ago |
Megvii Engine Team
|
d2a1905ad5
|
Revert "feat(mgb): add cumprod opr"
This reverts commit 3436c3bdaa .
GitOrigin-RevId: 95ab3d1aa7
|
2 years ago |
Megvii Engine Team
|
49e14f87b5
|
feat(mgb): add cumprod opr
GitOrigin-RevId: 3436c3bdaa
|
3 years ago |
Megvii Engine Team
|
c49d3070ba
|
refactor(imperative/ops): extends DnnOprCaller with template
GitOrigin-RevId: 402cba209a
|
3 years ago |
Megvii Engine Team
|
7b17c1180e
|
refactor(dnn): make cudnn_frontend work
GitOrigin-RevId: f089f93494
|
3 years ago |
Megvii Engine Team
|
35e9cc9845
|
feat(dnn/cuda): add cudnn frontend api
GitOrigin-RevId: 9b18a57893
|
3 years ago |
Megvii Engine Team
|
2886245bb1
|
perf(imperative/src): improve pad host performance
GitOrigin-RevId: 05223deca7
|
3 years ago |
Megvii Engine Team
|
b55942a94d
|
feat(dnn/naive/norm,-dnn/cuda/norm,-dnn/test/norm): add norm dnn opr,
fwd only
GitOrigin-RevId: 989474168d
|
3 years ago |
Megvii Engine Team
|
24c5c19bf0
|
fix(imperative): make functional ops support negative axis
GitOrigin-RevId: f61e01270b
|
3 years ago |
Megvii Engine Team
|
8f17b84ad8
|
fix(dnn): fix dnn run cd4 on cpu
GitOrigin-RevId: 5eae7496e5
|
3 years ago |
Megvii Engine Team
|
96d90be1c6
|
feat(dnn): fallback support int4 relayout
GitOrigin-RevId: 3625f58470
|
3 years ago |
Megvii Engine Team
|
da91e650a5
|
refactor(ops/layer_norm): speed up the host speed of layer_norm
GitOrigin-RevId: 6f359b5b29
|
3 years ago |
Megvii Engine Team
|
98b5ee78c1
|
feat(mge/dnn): add lamb optimizer
GitOrigin-RevId: 5a27157456
|
3 years ago |
Megvii Engine Team
|
c2e9860feb
|
chore(license): remove all license in file header
GitOrigin-RevId: a0e31247a6
|
3 years ago |
Megvii Engine Team
|
48526abb79
|
fix(mgb): fix concat cd4 tensor check size invalid
GitOrigin-RevId: 065e0b4be0
|
3 years ago |
Megvii Engine Team
|
597a1e791b
|
refactor(imperative): add interface to clear algorithm cache
GitOrigin-RevId: 662618954b
|
3 years ago |
Megvii Engine Team
|
e2f5156b69
|
refactor(megbrain): save fastrun result to algorithm cache
GitOrigin-RevId: 45301ebb4d
|
3 years ago |
Megvii Engine Team
|
409c988163
|
fix(imperative): add matmul apply_on_varnode
GitOrigin-RevId: 2cf6bf237c
|
3 years ago |
Megvii Engine Team
|
73112558d0
|
feat(mge/dnn): support checknonfinite for fp16
GitOrigin-RevId: 83fa139ac0
|
3 years ago |
Megvii Engine Team
|
14e9ad625d
|
fix(megdnn): emit define-but-not-referenced and extra-;-ignored warning on cuda9.0~cuda9.1
GitOrigin-RevId: f6db42e395
|
3 years ago |
Megvii Engine Team
|
d2278f02d2
|
perf(imperative): speed up conv_transpose3d
GitOrigin-RevId: e741305446
|
3 years ago |
Megvii Engine Team
|
d9c4ef59fe
|
perf(imperative): using simple hash key in heuristic cache
GitOrigin-RevId: 6fddd612e7
|
3 years ago |
Megvii Engine Team
|
26ea33c6a7
|
perf(imperative): improve convbwd performance
GitOrigin-RevId: cfc8623d7a
|
3 years ago |
Megvii Engine Team
|
3949d425fb
|
feat(core): always show MegEngine version and git commit id
GitOrigin-RevId: 4daa5be6d6
|
3 years ago |
Megvii Engine Team
|
fd6f8e58b0
|
feat(mgb/dtype): add dtype qint1
GitOrigin-RevId: abe9fb68b1
|
3 years ago |
Megvii Engine Team
|
2a900a69cb
|
perf(imperative): improve reduce op performance
GitOrigin-RevId: 26d982a7b8
|
3 years ago |
Megvii Engine Team
|
72a70dd6a7
|
perf(imperative): specialize convolution implementation
GitOrigin-RevId: 33634c550f
|
3 years ago |
Megvii Engine Team
|
2b80806f21
|
perf(imperative/src): improve dot performance
GitOrigin-RevId: 35b5bd164f
|
3 years ago |
Megvii Engine Team
|
e400b7ffe5
|
perf(imperative): enable memory forwarding for imperative
GitOrigin-RevId: 7c1993979c
|
3 years ago |
Megvii Engine Team
|
95ac055538
|
feat(dnn,mgb,imperative): add diag opr implement
GitOrigin-RevId: 43016ffa2b
|
3 years ago |
Megvii Engine Team
|
cbbca5fb10
|
feat(mge): add softmax op use cudnn api
GitOrigin-RevId: 7734ebf8c4
|
3 years ago |
Megvii Engine Team
|
20b42a8c3b
|
fix(dnn): add naive lstm kernel
GitOrigin-RevId: f08ef810cf
|
3 years ago |
Megvii Engine Team
|
2faa6ea5a9
|
Merge pull request #213 from kxz18:rnn
GitOrigin-RevId: 9e9215c115
|
3 years ago |
Megvii Engine Team
|
1999307015
|
feat(mgb/opr): add dropout kernel
GitOrigin-RevId: d248bd2005
|
3 years ago |
Megvii Engine Team
|
a93741815b
|
feat(mgb/opr): add layernorm forward and backward kernel
GitOrigin-RevId: 0cd484e753
|
3 years ago |
Megvii Engine Team
|
93310c0e4b
|
fix(mgb/gopt): fix cpu global layout transform fastrun error
GitOrigin-RevId: ea254297e5
|
3 years ago |
kxz@thumt102-1
|
8f48da7ffe
|
feat(mgb/opr): add cell level rnn/lstm and sequence level rnn/lstm
|
3 years ago |
Megvii Engine Team
|
2881934cb8
|
feat(dnn/check_non_finite): addmul scale to check_non_finite opr
GitOrigin-RevId: c35a219e52
|
3 years ago |
Megvii Engine Team
|
11d75fecb5
|
feat(dnn/check_non_finite): add batch check_non_finite
GitOrigin-RevId: e108133282
|
3 years ago |
Megvii Engine Team
|
3d45d35241
|
feat(mgb/gopt): profiler support checking algo availability
GitOrigin-RevId: 39cad612cc
|
3 years ago |
Megvii Engine Team
|
d90cb7763c
|
feat(src/core): record support change ptr basic
GitOrigin-RevId: b12f1c4a66
|
3 years ago |
Megvii Engine Team
|
68cdabd288
|
feat(opr): indexing_multi_axis_vec support nd index
GitOrigin-RevId: 07b1248bdc
|
3 years ago |