luzzyzhang
|
16f052e916
|
fix(megdnn): change ver 60 to use cuda capability 50
fix(megdnn): if compute capability less than 60 skip fp16 test
|
5 years ago |
Megvii Engine Team
|
c4dfdbd26d
|
refactor(dnn/fallback): refactor im2col
GitOrigin-RevId: b58770211e
|
5 years ago |
Megvii Engine Team
|
23478a0d53
|
test(dnn/cuda): fix cuda int8 test on sm60
GitOrigin-RevId: 66bab333e1
|
5 years ago |
Megvii Engine Team
|
5121626e98
|
refactor(dnn/parampack): remove parampacksplit
GitOrigin-RevId: ed00341d58
|
5 years ago |
Megvii Engine Team
|
b708f15d39
|
refactor(mgb/param_pack): use shared mem for param pack
GitOrigin-RevId: bc56f09037
|
5 years ago |
Megvii Engine Team
|
712b87c8c1
|
feat(mgb/core): add comp node for cambricon
add testcase for cambricon comp node
GitOrigin-RevId: 7794faa47f
|
5 years ago |
Megvii Engine Team
|
ca811c2c91
|
fix(mgb/opr/reduce): fix grad function of reduce mean and add testcase
GitOrigin-RevId: 390854bb2f
|
5 years ago |
Megvii Engine Team
|
ec9de227b6
|
fix(mgb/dnn): fix opencl local opr
GitOrigin-RevId: 38e0a658a2
|
5 years ago |
Megvii Engine Team
|
f5833a5294
|
fix(dnn/cuda): fix cublas matmul on sm60
GitOrigin-RevId: 3fc0c30a23
|
5 years ago |
Megvii Engine Team
|
90ca85541e
|
feat(dnn/x86): add avx2 int8 stride1 chanwise multithread conv
GitOrigin-RevId: 8f310c3d13
|
5 years ago |
Megvii Engine Team
|
8ba8c11d87
|
feat(dnn): add nchw44 layout
GitOrigin-RevId: d92672b88a
|
5 years ago |
Megvii Engine Team
|
f91881ffdc
|
MegEngine: Initial commit of MegEngine.
GitOrigin-RevId: f0c8338beb
|
5 years ago |