Megvii Engine Team
|
35e9cc9845
|
feat(dnn/cuda): add cudnn frontend api
GitOrigin-RevId: 9b18a57893
|
3 years ago |
Megvii Engine Team
|
c2e9860feb
|
chore(license): remove all license in file header
GitOrigin-RevId: a0e31247a6
|
3 years ago |
Megvii Engine Team
|
e2f5156b69
|
refactor(megbrain): save fastrun result to algorithm cache
GitOrigin-RevId: 45301ebb4d
|
3 years ago |
Megvii Engine Team
|
b6ad457269
|
feat(cuda): support int1 simplewq conv
GitOrigin-RevId: 9c37c41bc7
|
3 years ago |
Megvii Engine Team
|
3228fb75a5
|
fix(cuda): conv algo heuristic choose
GitOrigin-RevId: 95c5e7d627
|
3 years ago |
Megvii Engine Team
|
f7994683bd
|
feat(cuda): add large kernel direct conv to heuristic algo chooser
GitOrigin-RevId: bc927b6df7
|
3 years ago |
Megvii Engine Team
|
47fe766310
|
feat(dnn/cuda): add implicit bmm kernels for large kernel depthwise convolution backward filter opr
GitOrigin-RevId: 932e7689e8
|
3 years ago |
Megvii Engine Team
|
dcc9693582
|
feat(dnn/cuda): add heuristic rule for implicit batched gemm large kernel dwconv2d kernels
GitOrigin-RevId: 2d2c213bfd
|
3 years ago |
Megvii Engine Team
|
369c2ccc5a
|
style(all): reformat c++ code
GitOrigin-RevId: 3ffd1b211f
|
3 years ago |
Megvii Engine Team
|
de363c04af
|
Revert "perf(cuda/conv): cache serval cudnn api"
This reverts commit 188c62cdd6 .
GitOrigin-RevId: 92a82b8cd9
|
3 years ago |
Megvii Engine Team
|
d69b59035d
|
feat(dnn): add an get_all_algorithms_safe interface
GitOrigin-RevId: e3734e4531
|
3 years ago |
Megvii Engine Team
|
694aa1bd92
|
feat(dnn): add heuristic cache
GitOrigin-RevId: 35e942b5e3
|
3 years ago |
Megvii Engine Team
|
2aba0378b9
|
refactor(mgb/dnn): fix group conv is_available
GitOrigin-RevId: b279909168
|
3 years ago |
Megvii Engine Team
|
6ce212d2e0
|
refactor(mgb): refactor group conv
GitOrigin-RevId: 7afd312690
|
4 years ago |
Megvii Engine Team
|
7b2a76d1ee
|
refactor(mgb): make conv handle noncontiguous tensors
GitOrigin-RevId: 86282709b3
|
4 years ago |
Megvii Engine Team
|
938944027d
|
fix(mgb/dnn): fix cudnn8 convbias
GitOrigin-RevId: 0fdbfd258c
|
4 years ago |
Megvii Engine Team
|
5419a95d1e
|
perf(cuda/conv): cache serval cudnn api
GitOrigin-RevId: 188c62cdd6
|
4 years ago |
Megvii Engine Team
|
c338e876ec
|
refactor(mgb/dnn): add negative attribute for algo
GitOrigin-RevId: 88b1ce94a5
|
4 years ago |
Megvii Engine Team
|
ec1a99acc2
|
refactor(mgb/dnn): replace reproducible with attribute
GitOrigin-RevId: d49015714c
|
4 years ago |
Megvii Engine Team
|
c82d88751a
|
fix(dnn/cuda): add cuda nchw int8 conv impl with nchw4 to fix cu111 compatibility
GitOrigin-RevId: 771968f9ac
|
4 years ago |
Megvii Engine Team
|
e4b71bdf64
|
refactor(megdnn): remove unnessary 1x1 algo
GitOrigin-RevId: 809f9b2bba
|
4 years ago |
Megvii Engine Team
|
364afec033
|
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
|
4 years ago |
Megvii Engine Team
|
142f31a875
|
perf(dnn/cuda): change conv_bias heu, prefer dnn chanwise impl, dislike dnn batch gemm conv1x1
GitOrigin-RevId: 323bf6073a
|
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
|
60c6d59fc9
|
feat(mbg/core): support bias preprocess in conv_bias
GitOrigin-RevId: d2e1e14d41
|
4 years ago |
Megvii Engine Team
|
4aa277a203
|
refactor(dnn/cuda): misc
GitOrigin-RevId: 1f8f91a0cc
|
4 years ago |
Megvii Engine Team
|
b8b000db3b
|
feat(dnn/fallback): fix fallback interface of weight preprocess
GitOrigin-RevId: ca860f487e
|
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
|
0293d58ade
|
feat(mge): add bfloat16 support
GitOrigin-RevId: a942ce6791
|
5 years ago |
Megvii Engine Team
|
f91881ffdc
|
MegEngine: Initial commit of MegEngine.
GitOrigin-RevId: f0c8338beb
|
5 years ago |