Megvii Engine Team
|
8e5410e41f
|
feat(cuda): add fp16 compute 16 kernel
GitOrigin-RevId: e03435be02
|
3 years ago |
Megvii Engine Team
|
472e2f9655
|
refactor(cuda): depthwish large kernel
GitOrigin-RevId: dade8710b4
|
3 years ago |
Megvii Engine Team
|
e698ec20c2
|
feat(cuda): float16 depthwise large kernel conv compute fp32
GitOrigin-RevId: 3050d48f26
|
3 years ago |
Megvii Engine Team
|
48406382ce
|
feat(cuda): support float16 depthwise large kernel conv
GitOrigin-RevId: fdc1b15fbc
|
3 years ago |
Megvii Engine Team
|
7042f76b34
|
perf(cuda): speedup conv backward data with small feature map and large filter size
GitOrigin-RevId: 85592bca6b
|
3 years ago |
Megvii Engine Team
|
87a2aeebb1
|
perf(cuda): speedup chanwise conv with small feature map and large filter size
GitOrigin-RevId: e65b2ce856
|
3 years ago |
Megvii Engine Team
|
369c2ccc5a
|
style(all): reformat c++ code
GitOrigin-RevId: 3ffd1b211f
|
3 years ago |
Megvii Engine Team
|
722aecd437
|
feat(mgb): support fp16 nhwc backward
GitOrigin-RevId: 954ac6405a
|
3 years ago |
Megvii Engine Team
|
4c13bc7e1b
|
feat(dnn/cuda): add nhwc int8 deconv
GitOrigin-RevId: ad361a0f81
|
3 years ago |
Megvii Engine Team
|
11f022ff7c
|
feat(dnn/cuda): add nhwc int8 imma conv and conv fuse typecvt
GitOrigin-RevId: 229e1eb4be
|
3 years ago |
Megvii Engine Team
|
d915c5a3fd
|
refactor(mgb): make convolution3D handle noncontiguous tensors
GitOrigin-RevId: 3d3c31b021
|
4 years ago |
Megvii Engine Team
|
d04cd67faf
|
refactor(mgb): make conv-backward-filter handle noncontiguous tensors
GitOrigin-RevId: 44c586f912
|
4 years ago |
Megvii Engine Team
|
44376f702a
|
refactor(mgb): make conv-backward-data handle noncontiguous tensors
GitOrigin-RevId: 0a8f66f9d3
|
4 years ago |
Megvii Engine Team
|
1e6ef3771f
|
feat(mgb/dnn): add accuracy shake checker
GitOrigin-RevId: 0bb52078a1
|
4 years ago |
Megvii Engine Team
|
ba2ad46e54
|
feat(gopt): add deconv nchw4 int8 opt pass, add deconv nchw int8
GitOrigin-RevId: c0530a949e
|
4 years ago |
Megvii Engine Team
|
5d350fc843
|
feat(dnn/cuda): add deconv int8 and fix cutlass conv wrapper base on modify cutlass 2.4
GitOrigin-RevId: 49e0565e8a
|
4 years ago |
Megvii Engine Team
|
b04ad06f84
|
refactor(megdnn): refactor matmul algo in conv backward filter
GitOrigin-RevId: bb65777583
|
4 years ago |
Megvii Engine Team
|
25089e520e
|
refactor(megdnn): refactor matmul algo in conv backward data
GitOrigin-RevId: 8de601df6e
|
4 years ago |
Megvii Engine Team
|
0d720653ac
|
refactor(megdnn): add default algo for convolution forward
GitOrigin-RevId: a12a7d399a
|
4 years ago |
Megvii Engine Team
|
659217acd2
|
refactor(megdnn): refactor bfloat16 convbias to recursive inteface
GitOrigin-RevId: 378194fb7f
|
4 years ago |
Megvii Engine Team
|
b8febaf91f
|
refactor(megdnn): refactor bfloat16 convolutionbackwardfilter to recursive inteface
GitOrigin-RevId: 37c08a5b8b
|
4 years ago |
Megvii Engine Team
|
f14e0c17e7
|
feat(mgb): add recursive for fastrun and megdnn test
GitOrigin-RevId: 743846f645
|
4 years ago |
Megvii Engine Team
|
364afec033
|
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
|
4 years ago |
Megvii Engine Team
|
a1877ee0fa
|
refactor(dnn): refactor algo interface, use algoinfo instead of global algorithm
GitOrigin-RevId: 479718ac75
|
4 years ago |
Megvii Engine Team
|
f354724220
|
fix(ci/megdnn_test/megbrain_test): split some
testcase, which caused by timeout
GitOrigin-RevId: ffed9d7820
|
5 years ago |
Megvii Engine Team
|
0293d58ade
|
feat(mge): add bfloat16 support
GitOrigin-RevId: a942ce6791
|
5 years ago |
Megvii Engine Team
|
1c4a64b2af
|
test(megdnn): skip fp16 test if compute capability less than 60
GitOrigin-RevId: 3377007b87
fix(dnn): fix include file redefined error
GitOrigin-RevId: 9d52f7ba50
|
5 years ago |
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
|
f5833a5294
|
fix(dnn/cuda): fix cublas matmul on sm60
GitOrigin-RevId: 3fc0c30a23
|
5 years ago |
Megvii Engine Team
|
f91881ffdc
|
MegEngine: Initial commit of MegEngine.
GitOrigin-RevId: f0c8338beb
|
5 years ago |