Megvii Engine Team
|
69fe5ab3b3
|
feat(dnn/cuda): add conv2d-sass-kernel
GitOrigin-RevId: f284d5a4ce
|
5 years ago |
Megvii Engine Team
|
9f352b1c45
|
feat(megbrain/dnn): add indexing remap int32 for naive and cuda
GitOrigin-RevId: 5f66d51de4
|
5 years ago |
Megvii Engine Team
|
32c86211ee
|
fix(dnn/cuda): enable cuda algos for nchw quantized
GitOrigin-RevId: 4d1e167b86
|
5 years ago |
Megvii Engine Team
|
0b32056828
|
feat(dnn/naive): support weight preprocess interface in dnn
GitOrigin-RevId: 84791aacf9
|
5 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 |
Megvii Engine Team
|
f582c1921e
|
fix(dnn/test): fix dnn assert and format code
GitOrigin-RevId: c684210c42
|
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
|
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
|
ca811c2c91
|
fix(mgb/opr/reduce): fix grad function of reduce mean and add testcase
GitOrigin-RevId: 390854bb2f
|
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 |