Megvii Engine Team
|
8da2f698a3
|
feat(dnn/cuda): support warp perspective/pooling op when channel not aligned to 64
GitOrigin-RevId: 39f29ec990
|
4 years ago |
Megvii Engine Team
|
4fe68ac9ed
|
feat(dnn/cuda): support transforming layout between nchw and nchw64 when channel not aligned to 64
GitOrigin-RevId: e9ecbcf2e2
|
4 years ago |
Megvii Engine Team
|
56e863b7d4
|
fix(dnn/cuda): fix int4 epilogue stg bug
GitOrigin-RevId: e86da9a8a8
|
4 years ago |
Megvii Engine Team
|
12a0e61542
|
feat(dnn/cuda): add cuda elemwise int4
GitOrigin-RevId: 8a9aaec328
|
4 years ago |
Megvii Engine Team
|
df1af59b5c
|
feat(dnn): warp perspective support int4
GitOrigin-RevId: 826a43b349
|
4 years ago |
Megvii Engine Team
|
2398df079c
|
feat(dnn/cuda): add cuda int4 pooling
GitOrigin-RevId: 14ed4e6f00
|
4 years ago |
Megvii Engine Team
|
e250afb08f
|
feat(dnn/cuda): support conv_bias for nchw64 and qint4
GitOrigin-RevId: 1c65ba87d7
|
4 years ago |
Megvii Engine Team
|
3b9b87809d
|
refactor(dnn): refactor lowbit tensor format
GitOrigin-RevId: b646dc085b
|
4 years ago |
Megvii Engine Team
|
8fef78d06d
|
feat(dnn/cuda): add relayout format when width is an odd number
GitOrigin-RevId: f059f1f56d
|
4 years ago |
Megvii Engine Team
|
91d6160769
|
feat(dnn/common): add tensor format for low-bits tensor layout
GitOrigin-RevId: 0aa3753f37
|
4 years ago |
Megvii Engine Team
|
19a554d674
|
test(dnn/cuda): add testcase for transforming tensor layout between nchw and nchw64
GitOrigin-RevId: 75d579635a
|
4 years ago |
Megvii Engine Team
|
23032f50f2
|
feat(dnn/cuda): support float16 for index_incr_multi_axis_vec
GitOrigin-RevId: c2ae93d568
|
4 years ago |
Megvii Engine Team
|
938944027d
|
fix(mgb/dnn): fix cudnn8 convbias
GitOrigin-RevId: 0fdbfd258c
|
4 years ago |
Megvii Engine Team
|
1525a02530
|
feat(mge/module): add python wrapper for unfold
GitOrigin-RevId: 562103186f
|
4 years ago |
Megvii Engine Team
|
13b15fb08c
|
feat(megbrain): add correlation opr
GitOrigin-RevId: 6d44598891
|
4 years ago |
Megvii Engine Team
|
1997b1a289
|
feat(dnn/cuda): add correlation kernel
GitOrigin-RevId: 25e58b61e6
|
4 years ago |
Megvii Engine Team
|
c3f8cf04fa
|
feat(dnn): add conv_bwd_data and conv_bwd_filter accuracy shake check
GitOrigin-RevId: 4069e083d2
|
4 years ago |
Megvii Engine Team
|
1e6ef3771f
|
feat(mgb/dnn): add accuracy shake checker
GitOrigin-RevId: 0bb52078a1
|
4 years ago |
Megvii Engine Team
|
78fff72a95
|
feat(dnn): add param_pack for rocm
GitOrigin-RevId: 2180504c71
|
4 years ago |
Megvii Engine Team
|
ff755451d2
|
refactor(mgb): move algo's name from info to desc and delete some algo's unnecessary param() method
GitOrigin-RevId: 144ff547d1
|
4 years ago |
Megvii Engine Team
|
756c1eb7f2
|
fix(mgb/dnn): add cuda float naive matmul algo
GitOrigin-RevId: db7f7fc057
|
4 years ago |
Megvii Engine Team
|
a49f4a66b7
|
feat(dnn): add indexing_one_hot and indexing_set_one_hot opr
GitOrigin-RevId: c5406c71ff
|
4 years ago |
Megvii Engine Team
|
2fd3fa8834
|
feat(cmake): update for enflame cmake compile
GitOrigin-RevId: 3c3c6b3462
|
4 years ago |
Megvii Engine Team
|
68f2e59763
|
fix(mgb(ci)): fix tx1 ci testcase
GitOrigin-RevId: 7ac0c4b68d
|
4 years ago |
Megvii Engine Team
|
55974e8cf9
|
feat(log): opt log
* opt log at release mode
* add MGE_OVERRIDE_LOG_LEVEL for runtime debug
//! env to config LogLevel
//! DEBUG = 0, INFO = 1, WARN = 2, ERROR = 3, NO_LOG = 4
//! for example , export MGE_OVERRIDE_LOG_LEVEL=0, means set LogLevel to DEBUG
GitOrigin-RevId: 16cd674c56
|
4 years ago |
Megvii Engine Team
|
1c01128ff5
|
fix SASS cu111
GitOrigin-RevId: cb075ad154
|
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
|
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
|
f2b42bf09e
|
chore(dotprod): add arm dotprod attribute for easy use
GitOrigin-RevId: 78c3e72218
|
4 years ago |
Megvii Engine Team
|
97beae2fd8
|
fix(megdnn): fix megdnn benchmark testcase
GitOrigin-RevId: 726971474a
|
4 years ago |
Megvii Engine Team
|
2de2222e46
|
feat(dnn/cuda): add cutlass batched gemv kernel for matmul operator
GitOrigin-RevId: 51702c4e79
|
4 years ago |
Megvii Engine Team
|
973d2a0ac2
|
feat(dnn/cuda): add cutlass matmul using split k parallel
GitOrigin-RevId: 650209e35f
|
4 years ago |
Megvii Engine Team
|
03c921f7c4
|
feat(dnn/cuda): add cutlass matmul impls
GitOrigin-RevId: 619c8c299c
|
4 years ago |
Megvii Engine Team
|
5b62acfa01
|
feat(dnn/armv7): add new matmul strategy k8x8x4
GitOrigin-RevId: 0c6b7fa1b2
|
4 years ago |
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
|
0e3a6329ff
|
build(cuda): support cu111 build
GitOrigin-RevId: b3067ba4d5
|
4 years ago |
Megvii Engine Team
|
cd7090acbb
|
fix(opencl): enable image on mali(cl2.1)
GitOrigin-RevId: 0c670fba80
|
4 years ago |
Megvii Engine Team
|
c51a687cef
|
chore(mge): update copyright years
GitOrigin-RevId: 46104ac891
|
4 years ago |
Megvii Engine Team
|
af42ce7e69
|
fix(megdnn): some fixes of execution policy
GitOrigin-RevId: 920f39bcb6
|
4 years ago |
Megvii Engine Team
|
821656aa4b
|
refactor(megdnn): refactor brute force algo in batched matmul
GitOrigin-RevId: 5c143ab3ac
|
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
|
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 |