Megvii Engine Team
|
b6ad457269
|
feat(cuda): support int1 simplewq conv
GitOrigin-RevId: 9c37c41bc7
|
4 years ago |
Megvii Engine Team
|
331567af5d
|
fix(opencl/ci): misc opt and fix:
1: fix megbrain test failed on mali 2.1 devices
2: reduce ci time (about reduce 20min)
GitOrigin-RevId: 4dcdcd48a6
|
4 years ago |
Megvii Engine Team
|
fd6f8e58b0
|
feat(mgb/dtype): add dtype qint1
GitOrigin-RevId: abe9fb68b1
|
4 years ago |
Megvii Engine Team
|
2b80806f21
|
perf(imperative/src): improve dot performance
GitOrigin-RevId: 35b5bd164f
|
4 years ago |
Megvii Engine Team
|
3c3fc6f33c
|
refactor(imperative): move python code of elemwise/reduce/conv2d/bn to c++
GitOrigin-RevId: 01b5324392
|
4 years ago |
Megvii Engine Team
|
e400b7ffe5
|
perf(imperative): enable memory forwarding for imperative
GitOrigin-RevId: 7c1993979c
|
4 years ago |
Megvii Engine Team
|
1ce78aa09b
|
fix(imperative): destruct dnn handles at last
GitOrigin-RevId: 7a67c68c55
|
4 years ago |
Megvii Engine Team
|
3726f5cc92
|
feat(gopt): merger consecutive relayout and dimshuffle to one relayout to optimize CD4 performarce
GitOrigin-RevId: a058776be3
|
4 years ago |
Megvii Engine Team
|
04193e3bd1
|
feat(dnn): add nearest mode for remap and resize
GitOrigin-RevId: 31e7b72a78
|
4 years ago |
Megvii Engine Team
|
95ac055538
|
feat(dnn,mgb,imperative): add diag opr implement
GitOrigin-RevId: 43016ffa2b
|
4 years ago |
Megvii Engine Team
|
cbbca5fb10
|
feat(mge): add softmax op use cudnn api
GitOrigin-RevId: 7734ebf8c4
|
4 years ago |
Megvii Engine Team
|
20b42a8c3b
|
fix(dnn): add naive lstm kernel
GitOrigin-RevId: f08ef810cf
|
4 years ago |
Megvii Engine Team
|
2faa6ea5a9
|
Merge pull request #213 from kxz18:rnn
GitOrigin-RevId: 9e9215c115
|
4 years ago |
Megvii Engine Team
|
3b41840b68
|
fix(mgb): change caffepooling log level
GitOrigin-RevId: 290d243ef5
|
4 years ago |
Megvii Engine Team
|
1999307015
|
feat(mgb/opr): add dropout kernel
GitOrigin-RevId: d248bd2005
|
4 years ago |
Megvii Engine Team
|
a93741815b
|
feat(mgb/opr): add layernorm forward and backward kernel
GitOrigin-RevId: 0cd484e753
|
4 years ago |
Megvii Engine Team
|
f6d9909460
|
feat(dnn): add elemwise multi type support i16xf32 and u8xf32
GitOrigin-RevId: 2fe469bb4e
|
4 years ago |
Megvii Engine Team
|
d9a46ea47b
|
fix(dnn): correct behaviour of floor div for int tensor
GitOrigin-RevId: 1444f69cce
|
4 years ago |
Megvii Engine Team
|
0ad5eeaedd
|
feat(mgb/gopt): global layout transform support opencl
GitOrigin-RevId: 132605c7d9
|
4 years ago |
kxz@thumt102-1
|
8f48da7ffe
|
feat(mgb/opr): add cell level rnn/lstm and sequence level rnn/lstm
|
4 years ago |
Megvii Engine Team
|
2881934cb8
|
feat(dnn/check_non_finite): addmul scale to check_non_finite opr
GitOrigin-RevId: c35a219e52
|
4 years ago |
Megvii Engine Team
|
6bb5409976
|
feat(dnn/src): add images2neibs kernel of opencl and related test
GitOrigin-RevId: 82242b7437
|
4 years ago |
Megvii Engine Team
|
6ce4a34403
|
feat(dnn): add fallback postprocess
GitOrigin-RevId: 4201a0f158
|
4 years ago |
Megvii Engine Team
|
2696e4efaa
|
feat(dnn): add float16 for remap backward
GitOrigin-RevId: 0263030051
|
4 years ago |
Megvii Engine Team
|
11d75fecb5
|
feat(dnn/check_non_finite): add batch check_non_finite
GitOrigin-RevId: e108133282
|
4 years ago |
Megvii Engine Team
|
30976c239f
|
fix(mgb/gopt): fix global layout transform
add a special opr_format modify function for concat operators to modify concat axis when input's layout has been changed
GitOrigin-RevId: 4094208057
|
4 years ago |
Megvii Engine Team
|
ca7cec7a5d
|
fix(mgb/gopt): minor fixes for global layout transform
Merge Target::ARM and Target::X86 into Target::CPU to make global layout transform easier to use
GitOrigin-RevId: cc9363fa38
|
4 years ago |
Megvii Engine Team
|
fe93013a6e
|
feat(mgb/gopt): global layout transform support nchw_nchwxx hybrid mode
GitOrigin-RevId: 6d5b55d7fc
|
4 years ago |
Megvii Engine Team
|
c85631aa77
|
feat(dnn): use ref ptr interface for all backends
GitOrigin-RevId: f65feae5cc
|
4 years ago |
Megvii Engine Team
|
d90cb7763c
|
feat(src/core): record support change ptr basic
GitOrigin-RevId: b12f1c4a66
|
4 years ago |
Megvii Engine Team
|
89186edc5d
|
fix(dnn): correct reduce/argmxx/fakequant calculation with nan
GitOrigin-RevId: 7e78bdae91
|
4 years ago |
Megvii Engine Team
|
68cdabd288
|
feat(opr): indexing_multi_axis_vec support nd index
GitOrigin-RevId: 07b1248bdc
|
4 years ago |
Megvii Engine Team
|
c48d58daa8
|
feat(dnn/arm_common): add N1HW like elemwise broadcast mode
GitOrigin-RevId: 2895135801
|
4 years ago |
Megvii Engine Team
|
26634db7a8
|
fix(dnn): support relayout for non-contigous layout
GitOrigin-RevId: 44a0adddba
|
4 years ago |
Megvii Engine Team
|
056fd6bc59
|
feat(dnn/arm64): support stride_m in arm64 relayout
GitOrigin-RevId: c74193a23d
|
4 years ago |
Megvii Engine Team
|
c50858ee13
|
fix(dnn): specialize pow to make it consistent
GitOrigin-RevId: cff3bbbadd
|
4 years ago |
Megvii Engine Team
|
b5bf56e0ee
|
style(dnn): add bypass of clang-format for dnn foreach_opr macro
GitOrigin-RevId: 2f4bfcff42
|
4 years ago |
Megvii Engine Team
|
5885b137fa
|
feat(dnn/arm): support layout like NHWC channel like broadcast on arm
GitOrigin-RevId: fb4300004c
|
4 years ago |
Megvii Engine Team
|
369c2ccc5a
|
style(all): reformat c++ code
GitOrigin-RevId: 3ffd1b211f
|
4 years ago |
Megvii Engine Team
|
bfb30dcb81
|
chore(format): fix compile bugs after code format
GitOrigin-RevId: 11a4b06f6f
|
4 years ago |
Megvii Engine Team
|
177dec94c5
|
feat(mgb/opr): add bgr2gray mode for cvtcolor opr
GitOrigin-RevId: d50415b236
|
4 years ago |
Megvii Engine Team
|
f5cb21ed3a
|
fix(mgb/opr): add non finite check
GitOrigin-RevId: a9fcd0a350
|
4 years ago |
Megvii Engine Team
|
2fc7358517
|
Revert "feat(dnn/apicache): add generic apicache"
This reverts commit 40b8ac2ab6.
GitOrigin-RevId: 0f9b2df4c8
|
4 years ago |
Megvii Engine Team
|
de363c04af
|
Revert "perf(cuda/conv): cache serval cudnn api"
This reverts commit 188c62cdd6.
GitOrigin-RevId: 92a82b8cd9
|
4 years ago |
Megvii Engine Team
|
729ee64988
|
Revert "fix(api_cache): lock api cache for thread safety"
This reverts commit 8a244677c3.
GitOrigin-RevId: 582488adeb
|
4 years ago |
Megvii Engine Team
|
64c922c4bb
|
Revert "fix(api_cache): fix serialization for conv_desc"
This reverts commit 95dbc9c685.
GitOrigin-RevId: ca8c67b6b3
|
4 years ago |
Megvii Engine Team
|
b3e54eade1
|
feat(dnn/bn): use new cudnn BN kernel to support NHWC
GitOrigin-RevId: 9d80f2009d
|
4 years ago |
Megvii Engine Team
|
3977b7aa0b
|
feat(mgb/shuffle): add shuffle opr
GitOrigin-RevId: 80490a6f84
|
4 years ago |
Megvii Engine Team
|
17371e79b9
|
fix(dnn/reduce): fix reduce_mean o16c32 is incorrect for large tensor
GitOrigin-RevId: ebf03d814a
|
4 years ago |
Megvii Engine Team
|
8a3eb05a1b
|
refactor(mgb/gopt): refactor tensor reformat opt pass
GitOrigin-RevId: a1b1e89b76
|
4 years ago |