Megvii Engine Team
|
cf27dd642c
|
fix(cuda): use cudnn8.0.4 as cu111 default libs
GitOrigin-RevId: 721ca73bae
|
4 years ago |
Megvii Engine Team
|
649e4dd750
|
test(cuda): fix test for cu111
GitOrigin-RevId: 04fe5eb23f
|
4 years ago |
Megvii Engine Team
|
c69359d00d
|
fix(dnn/cuda): disable cudnn conv_bias kernels for NCHW4_NCHW tensor format
due to illegal memory access errors caused by the kernels
GitOrigin-RevId: 584cedeeab
|
4 years ago |
Megvii Engine Team
|
08ff62deb6
|
refactor(megdnn): refactor batched matmul algo in conv bias
GitOrigin-RevId: 64fda611ff
|
4 years ago |
Megvii Engine Team
|
8773926ef8
|
refactor(megdnn): refactor matmul algo in conv bias
GitOrigin-RevId: 932f7d6f81
|
4 years ago |
Megvii Engine Team
|
e4b71bdf64
|
refactor(megdnn): remove unnessary 1x1 algo
GitOrigin-RevId: 809f9b2bba
|
4 years ago |
Megvii Engine Team
|
659217acd2
|
refactor(megdnn): refactor bfloat16 convbias to recursive inteface
GitOrigin-RevId: 378194fb7f
|
4 years ago |
Megvii Engine Team
|
364afec033
|
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
|
4 years ago |
Megvii Engine Team
|
6856ce9ce2
|
feat(dnn): support conv bias activation for nchw4 input tensor format and nchw output tensor format
GitOrigin-RevId: 29cd73f87b
|
4 years ago |
Megvii Engine Team
|
32c86211ee
|
fix(dnn/cuda): enable cuda algos for nchw quantized
GitOrigin-RevId: 4d1e167b86
|
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
|
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 |