Megvii Engine Team
|
f60ab501ef
|
fix(mgb/opt): nchw to nchw4 pass suppport ic less than 4
GitOrigin-RevId: a3c205f38f
|
6 years ago |
Megvii Engine Team
|
28d85838ef
|
feat(dnn): add relayout_format for nchw to nchw4 and ic <=4
GitOrigin-RevId: 07f2ee6c5b
|
6 years ago |
Megvii Engine Team
|
43b42a651c
|
fix(dnn/cuda): fix indexing logic in psroi_pooling
a variable relating to indexing was not computed correctly
GitOrigin-RevId: 548c8f3f14
|
6 years ago |
Megvii Engine Team
|
53c288a304
|
fix(dnn/cuda): fix topk grid oversize
GitOrigin-RevId: d3c811a034
|
6 years ago |
Megvii Engine Team
|
9f352b1c45
|
feat(megbrain/dnn): add indexing remap int32 for naive and cuda
GitOrigin-RevId: 5f66d51de4
|
6 years ago |
Megvii Engine Team
|
32c86211ee
|
fix(dnn/cuda): enable cuda algos for nchw quantized
GitOrigin-RevId: 4d1e167b86
|
6 years ago |
Megvii Engine Team
|
0b32056828
|
feat(dnn/naive): support weight preprocess interface in dnn
GitOrigin-RevId: 84791aacf9
|
6 years ago |
Megvii Engine Team
|
0293d58ade
|
feat(mge): add bfloat16 support
GitOrigin-RevId: a942ce6791
|
6 years ago |
Megvii Engine Team
|
1255c9f13d
|
feat(mge/opr): add opr remap in opencl and naive
GitOrigin-RevId: 4540788660
|
6 years ago |
Megvii Engine Team
|
d6b098a052
|
refactor(dnn/cuda): remove function name weiming
GitOrigin-RevId: f36495a46a
|
6 years ago |
Megvii Engine Team
|
ca24c4cdd8
|
refactor(dnn/cuda): invoke local opr in group local
GitOrigin-RevId: aa636041bb
|
6 years ago |
Megvii Engine Team
|
91c3d5fe15
|
fix(dnn/cuda): add sharemem to prefer and refactor code
GitOrigin-RevId: 0dba39f56b
|
6 years ago |
Megvii Engine Team
|
e3de2011c0
|
build(mge): clang compile and link error
GitOrigin-RevId: 7062b451aa
|
6 years ago |
Megvii Engine Team
|
5121626e98
|
refactor(dnn/parampack): remove parampacksplit
GitOrigin-RevId: ed00341d58
|
6 years ago |
Megvii Engine Team
|
37b67c9b2b
|
refactor(dnn/parampack): reduce param pack memory use
GitOrigin-RevId: a802a14e8d
|
6 years ago |
Megvii Engine Team
|
ca811c2c91
|
fix(mgb/opr/reduce): fix grad function of reduce mean and add testcase
GitOrigin-RevId: 390854bb2f
|
6 years ago |
Megvii Engine Team
|
f5833a5294
|
fix(dnn/cuda): fix cublas matmul on sm60
GitOrigin-RevId: 3fc0c30a23
|
6 years ago |
Megvii Engine Team
|
f91881ffdc
|
MegEngine: Initial commit of MegEngine.
GitOrigin-RevId: f0c8338beb
|
6 years ago |