Megvii Engine Team
|
3f5238fb38
|
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 |