diff options
author | Cedric Nugteren <web@cedricnugteren.nl> | 2016-09-25 14:48:34 +0200 |
---|---|---|
committer | Cedric Nugteren <web@cedricnugteren.nl> | 2016-09-25 14:48:34 +0200 |
commit | 73d135c2cef9763b47d410b125eb8bb89ece8432 (patch) | |
tree | bc7aa23b620f717b04a348444f2a50164fa2f169 /src/kernels/level3/xgemm_direct.opencl | |
parent | 669f43aed65ccd4aae9c4a478e994660f3e2a592 (diff) |
Added a first version of a tuner for the GEMM direct kernel; collapsed MWGD, NWGD and KWGD into one WGD parameter
Diffstat (limited to 'src/kernels/level3/xgemm_direct.opencl')
-rw-r--r-- | src/kernels/level3/xgemm_direct.opencl | 186 |
1 files changed, 90 insertions, 96 deletions
diff --git a/src/kernels/level3/xgemm_direct.opencl b/src/kernels/level3/xgemm_direct.opencl index 801887dd..705ced9c 100644 --- a/src/kernels/level3/xgemm_direct.opencl +++ b/src/kernels/level3/xgemm_direct.opencl @@ -19,14 +19,8 @@ R"( // Parameters set by the tuner or by the database. Here they are given a basic default value in case // this kernel file is used outside of the CLBlast library. Note that all parameters here have a // suffix 'D' to denote that they are for the 'direct' version of the GEMM kernel. -#ifndef MWGD - #define MWGD 8 // Tile-size in dimension M (e.g. 64, 128) -#endif -#ifndef NWGD - #define NWGD 8 // Tile-size in dimension N (e.g. 64, 128) -#endif -#ifndef KWGD - #define KWGD 8 // Tile-size in dimension K (e.g. 8, 16) +#ifndef WGD + #define WGD 8 // Tile-size in dimension M, N, and K (e.g. 8, 16, 32, 64) #endif #ifndef MDIMCD #define MDIMCD 8 // Threads per workgroup in M-dimension (e.g. 8, 16, 32) @@ -41,7 +35,7 @@ R"( #define NDIMBD 8 // Re-shaped tile dimension of matrix B: KDIMBD * NDIMBD #endif #ifndef KWID - #define KWID 1 // Unroll factor of the KWGD loop (smaller or equal than KWGD) + #define KWID 1 // Unroll factor of the WGD loop (smaller or equal than WGD) #endif #ifndef VWMD #define VWMD 1 // Vector width of matrices A and C @@ -51,14 +45,14 @@ R"( #endif // Helper parameters based on the above tuning parameters -#define MWID (MWGD/MDIMCD) // Work per work-item (M-dimension) -#define NWID (NWGD/NDIMCD) // Work per work-item (N-dimension) +#define MWID (WGD/MDIMCD) // Work per work-item (M-dimension) +#define NWID (WGD/NDIMCD) // Work per work-item (N-dimension) #define KDIMAD ((MDIMCD*NDIMCD)/(MDIMAD)) // Re-shaped tile dimension of matrix A: KDIMAD * MDIMAD #define KDIMBD ((MDIMCD*NDIMCD)/(NDIMBD)) // Re-shaped tile dimension of matrix B: KDIMBD * NDIMBD -#define MWAD (MWGD/MDIMAD) // Amount of loads-per-thread for matrix A (M-dimension) -#define KWAD (KWGD/KDIMAD) // Amount of loads-per-thread for matrix A (K-dimension) -#define KWBD (KWGD/KDIMBD) // Amount of loads-per-thread for matrix B (K-dimension) -#define NWBD (NWGD/NDIMBD) // Amount of loads-per-thread for matrix B (N-dimension) +#define MWAD (WGD/MDIMAD) // Amount of loads-per-thread for matrix A (M-dimension) +#define KWAD (WGD/KDIMAD) // Amount of loads-per-thread for matrix A (K-dimension) +#define KWBD (WGD/KDIMBD) // Amount of loads-per-thread for matrix B (K-dimension) +#define NWBD (WGD/NDIMBD) // Amount of loads-per-thread for matrix B (N-dimension) // ================================================================================================= @@ -105,51 +99,51 @@ inline void GlobalToLocalDirectA(const __global realMD* restrict agm, __local re // Computes the indices for the global memory int mg = mia + la0*(MWAD/VWMD); int kg = kia + la1*KWAD; - int idm = (a_transpose) ? mg + kwg/VWMD : mg + GetGroupID0()*(MWGD/VWMD); - int idk = (a_transpose) ? kg + GetGroupID0()*MWGD : kg + kwg; + int idm = (a_transpose) ? mg + kwg/VWMD : mg + GetGroupID0()*(WGD/VWMD); + int idk = (a_transpose) ? kg + GetGroupID0()*WGD : kg + kwg; // Loads the data from global memory into the local memory const realMD avec = agm[idk*(a_ld/VWMD) + idm + a_offset]; #if VWMD == 1 - alm[kg*MWGD + mg] = avec; + alm[kg*WGD + mg] = avec; #elif VWMD == 2 - alm[kg*MWGD + mg*VWMD + 0] = avec.x; - alm[kg*MWGD + mg*VWMD + 1] = avec.y; + alm[kg*WGD + mg*VWMD + 0] = avec.x; + alm[kg*WGD + mg*VWMD + 1] = avec.y; #elif VWMD == 4 - alm[kg*MWGD + mg*VWMD + 0] = avec.x; - alm[kg*MWGD + mg*VWMD + 1] = avec.y; - alm[kg*MWGD + mg*VWMD + 2] = avec.z; - alm[kg*MWGD + mg*VWMD + 3] = avec.w; + alm[kg*WGD + mg*VWMD + 0] = avec.x; + alm[kg*WGD + mg*VWMD + 1] = avec.y; + alm[kg*WGD + mg*VWMD + 2] = avec.z; + alm[kg*WGD + mg*VWMD + 3] = avec.w; #elif VWMD == 8 - alm[kg*MWGD + mg*VWMD + 0] = avec.s0; - alm[kg*MWGD + mg*VWMD + 1] = avec.s1; - alm[kg*MWGD + mg*VWMD + 2] = avec.s2; - alm[kg*MWGD + mg*VWMD + 3] = avec.s3; - alm[kg*MWGD + mg*VWMD + 4] = avec.s4; - alm[kg*MWGD + mg*VWMD + 5] = avec.s5; - alm[kg*MWGD + mg*VWMD + 6] = avec.s6; - alm[kg*MWGD + mg*VWMD + 7] = avec.s7; + alm[kg*WGD + mg*VWMD + 0] = avec.s0; + alm[kg*WGD + mg*VWMD + 1] = avec.s1; + alm[kg*WGD + mg*VWMD + 2] = avec.s2; + alm[kg*WGD + mg*VWMD + 3] = avec.s3; + alm[kg*WGD + mg*VWMD + 4] = avec.s4; + alm[kg*WGD + mg*VWMD + 5] = avec.s5; + alm[kg*WGD + mg*VWMD + 6] = avec.s6; + alm[kg*WGD + mg*VWMD + 7] = avec.s7; #elif VWMD == 16 - alm[kg*MWGD + mg*VWMD + 0] = avec.s0; - alm[kg*MWGD + mg*VWMD + 1] = avec.s1; - alm[kg*MWGD + mg*VWMD + 2] = avec.s2; - alm[kg*MWGD + mg*VWMD + 3] = avec.s3; - alm[kg*MWGD + mg*VWMD + 4] = avec.s4; - alm[kg*MWGD + mg*VWMD + 5] = avec.s5; - alm[kg*MWGD + mg*VWMD + 6] = avec.s6; - alm[kg*MWGD + mg*VWMD + 7] = avec.s7; - alm[kg*MWGD + mg*VWMD + 8] = avec.s8; - alm[kg*MWGD + mg*VWMD + 9] = avec.s9; - alm[kg*MWGD + mg*VWMD + 10] = avec.sA; - alm[kg*MWGD + mg*VWMD + 11] = avec.sB; - alm[kg*MWGD + mg*VWMD + 12] = avec.sC; - alm[kg*MWGD + mg*VWMD + 13] = avec.sD; - alm[kg*MWGD + mg*VWMD + 14] = avec.sE; - alm[kg*MWGD + mg*VWMD + 15] = avec.sF; + alm[kg*WGD + mg*VWMD + 0] = avec.s0; + alm[kg*WGD + mg*VWMD + 1] = avec.s1; + alm[kg*WGD + mg*VWMD + 2] = avec.s2; + alm[kg*WGD + mg*VWMD + 3] = avec.s3; + alm[kg*WGD + mg*VWMD + 4] = avec.s4; + alm[kg*WGD + mg*VWMD + 5] = avec.s5; + alm[kg*WGD + mg*VWMD + 6] = avec.s6; + alm[kg*WGD + mg*VWMD + 7] = avec.s7; + alm[kg*WGD + mg*VWMD + 8] = avec.s8; + alm[kg*WGD + mg*VWMD + 9] = avec.s9; + alm[kg*WGD + mg*VWMD + 10] = avec.sA; + alm[kg*WGD + mg*VWMD + 11] = avec.sB; + alm[kg*WGD + mg*VWMD + 12] = avec.sC; + alm[kg*WGD + mg*VWMD + 13] = avec.sD; + alm[kg*WGD + mg*VWMD + 14] = avec.sE; + alm[kg*WGD + mg*VWMD + 15] = avec.sF; #endif if (a_conjugate) { for (int vm=0; vm<VWMD; ++vm) { - COMPLEX_CONJUGATE(alm[kg*MWGD + mg*VWMD + vm]); + COMPLEX_CONJUGATE(alm[kg*WGD + mg*VWMD + vm]); } } } @@ -170,51 +164,51 @@ inline void GlobalToLocalDirectB(const __global realND* restrict bgm, __local re // Computes the indices for the global memory int ng = nib + lb0*(NWBD/VWND); int kg = kib + lb1*KWBD; - int idn = (b_transpose) ? ng + kwg/VWND : ng + GetGroupID1()*(NWGD/VWND); - int idk = (b_transpose) ? kg + GetGroupID1()*NWGD : kg + kwg; + int idn = (b_transpose) ? ng + kwg/VWND : ng + GetGroupID1()*(WGD/VWND); + int idk = (b_transpose) ? kg + GetGroupID1()*WGD : kg + kwg; // Loads the data from global memory into the local memory - const realMD bvec = bgm[idk*(b_ld/VWND) + idn + b_offset]; + const realND bvec = bgm[idk*(b_ld/VWND) + idn + b_offset]; #if VWND == 1 - blm[kg*NWGD + ng] = bvec; + blm[kg*WGD + ng] = bvec; #elif VWND == 2 - blm[kg*NWGD + ng*VWND + 0] = bvec.x; - blm[kg*NWGD + ng*VWND + 1] = bvec.y; + blm[kg*WGD + ng*VWND + 0] = bvec.x; + blm[kg*WGD + ng*VWND + 1] = bvec.y; #elif VWND == 4 - blm[kg*NWGD + ng*VWND + 0] = bvec.x; - blm[kg*NWGD + ng*VWND + 1] = bvec.y; - blm[kg*NWGD + ng*VWND + 2] = bvec.z; - blm[kg*NWGD + ng*VWND + 3] = bvec.w; + blm[kg*WGD + ng*VWND + 0] = bvec.x; + blm[kg*WGD + ng*VWND + 1] = bvec.y; + blm[kg*WGD + ng*VWND + 2] = bvec.z; + blm[kg*WGD + ng*VWND + 3] = bvec.w; #elif VWND == 8 - blm[kg*NWGD + ng*VWND + 0] = bvec.s0; - blm[kg*NWGD + ng*VWND + 1] = bvec.s1; - blm[kg*NWGD + ng*VWND + 2] = bvec.s2; - blm[kg*NWGD + ng*VWND + 3] = bvec.s3; - blm[kg*NWGD + ng*VWND + 4] = bvec.s4; - blm[kg*NWGD + ng*VWND + 5] = bvec.s5; - blm[kg*NWGD + ng*VWND + 6] = bvec.s6; - blm[kg*NWGD + ng*VWND + 7] = bvec.s7; + blm[kg*WGD + ng*VWND + 0] = bvec.s0; + blm[kg*WGD + ng*VWND + 1] = bvec.s1; + blm[kg*WGD + ng*VWND + 2] = bvec.s2; + blm[kg*WGD + ng*VWND + 3] = bvec.s3; + blm[kg*WGD + ng*VWND + 4] = bvec.s4; + blm[kg*WGD + ng*VWND + 5] = bvec.s5; + blm[kg*WGD + ng*VWND + 6] = bvec.s6; + blm[kg*WGD + ng*VWND + 7] = bvec.s7; #elif VWND == 16 - blm[kg*NWGD + ng*VWND + 0] = bvec.s0; - blm[kg*NWGD + ng*VWND + 1] = bvec.s1; - blm[kg*NWGD + ng*VWND + 2] = bvec.s2; - blm[kg*NWGD + ng*VWND + 3] = bvec.s3; - blm[kg*NWGD + ng*VWND + 4] = bvec.s4; - blm[kg*NWGD + ng*VWND + 5] = bvec.s5; - blm[kg*NWGD + ng*VWND + 6] = bvec.s6; - blm[kg*NWGD + ng*VWND + 7] = bvec.s7; - blm[kg*NWGD + ng*VWND + 8] = bvec.s8; - blm[kg*NWGD + ng*VWND + 9] = bvec.s9; - blm[kg*NWGD + ng*VWND + 10] = bvec.sA; - blm[kg*NWGD + ng*VWND + 11] = bvec.sB; - blm[kg*NWGD + ng*VWND + 12] = bvec.sC; - blm[kg*NWGD + ng*VWND + 13] = bvec.sD; - blm[kg*NWGD + ng*VWND + 14] = bvec.sE; - blm[kg*NWGD + ng*VWND + 15] = bvec.sF; + blm[kg*WGD + ng*VWND + 0] = bvec.s0; + blm[kg*WGD + ng*VWND + 1] = bvec.s1; + blm[kg*WGD + ng*VWND + 2] = bvec.s2; + blm[kg*WGD + ng*VWND + 3] = bvec.s3; + blm[kg*WGD + ng*VWND + 4] = bvec.s4; + blm[kg*WGD + ng*VWND + 5] = bvec.s5; + blm[kg*WGD + ng*VWND + 6] = bvec.s6; + blm[kg*WGD + ng*VWND + 7] = bvec.s7; + blm[kg*WGD + ng*VWND + 8] = bvec.s8; + blm[kg*WGD + ng*VWND + 9] = bvec.s9; + blm[kg*WGD + ng*VWND + 10] = bvec.sA; + blm[kg*WGD + ng*VWND + 11] = bvec.sB; + blm[kg*WGD + ng*VWND + 12] = bvec.sC; + blm[kg*WGD + ng*VWND + 13] = bvec.sD; + blm[kg*WGD + ng*VWND + 14] = bvec.sE; + blm[kg*WGD + ng*VWND + 15] = bvec.sF; #endif if (b_conjugate) { for (int vn=0; vn<VWND; ++vn) { - COMPLEX_CONJUGATE(blm[kg*NWGD + ng*VWND + vn]); + COMPLEX_CONJUGATE(blm[kg*WGD + ng*VWND + vn]); } } } @@ -230,7 +224,7 @@ inline void LocalToPrivateDirectA(__local real* alm, real apm[MWID], const int k #pragma unroll for (int mi=0; mi<MWID; ++mi) { const int mg = mi + get_local_id(0)*MWID; - const int index = (a_transpose) ? mg*KWGD + kg : kg*MWGD + mg; + const int index = (a_transpose) ? mg*WGD + kg : kg*WGD + mg; apm[mi] = alm[index]; } } @@ -241,7 +235,7 @@ inline void LocalToPrivateDirectB(__local real* blm, real bpm[NWID], const int k #pragma unroll for (int ni=0; ni<NWID; ++ni) { const int ng = ni + get_local_id(1)*NWID; - const int index = (b_transpose) ? ng*KWGD + kg : kg*NWGD + ng; + const int index = (b_transpose) ? ng*WGD + kg : kg*WGD + ng; bpm[ni] = blm[index]; } } @@ -286,8 +280,8 @@ inline void StoreResultsDirect(__global real* cgm, real cpm[NWID][MWID], for (int mi=0; mi<MWID; ++mi) { int mg = mi + get_local_id(0)*MWID; int ng = ni + get_local_id(1)*NWID; - int idm = mg + GetGroupID0() * MWGD; - int idn = ng + GetGroupID1() * NWGD; + int idm = mg + GetGroupID0() * WGD; + int idn = ng + GetGroupID1() * WGD; // Determines the destination index const int c_index = (c_transpose) ? idm*c_ld + idn : idn*c_ld + idm; @@ -320,8 +314,8 @@ __kernel void XgemmDirect(const int kSizeM, const int kSizeN, const int kSizeK, const __global real* restrict bgms = (const __global real* restrict) bgm; // Allocates workgroup-private memory (local memory) - __local real alm[KWGD * MWGD]; - __local real blm[KWGD * NWGD]; + __local real alm[WGD * WGD]; + __local real blm[WGD * WGD]; // Combined thread identifier (volatile to disable caching) volatile int tid = get_local_id(0) + MDIMCD*get_local_id(1); @@ -335,15 +329,15 @@ __kernel void XgemmDirect(const int kSizeM, const int kSizeN, const int kSizeK, InitAccRegistersDirect(cpm); // The faster version of GEMM is not allowed on the (incomplete) borders. Therefore, this section - // processes only the main parts: output blocks of MWGD by NWGD. - const int idm = get_local_id(0) * MWID + GetGroupID0() * MWGD; - const int idn = get_local_id(1) * NWID + GetGroupID1() * NWGD; - if ((idm < (kSizeM/MWGD)*MWGD) && (idn < (kSizeN/NWGD)*NWGD) && + // processes only the main parts: output blocks of WGD by WGD. + const int idm = get_local_id(0) * MWID + GetGroupID0() * WGD; + const int idn = get_local_id(1) * NWID + GetGroupID1() * WGD; + if ((idm < (kSizeM/WGD)*WGD) && (idn < (kSizeN/WGD)*WGD) && (a_ld % VWMD == 0) && (b_ld % VWND == 0)) { // Loops over all complete workgroup tiles int kwg = 0; - for (; kwg < (kSizeK/KWGD) * KWGD; kwg+=KWGD) { + for (; kwg < (kSizeK/WGD) * WGD; kwg+=WGD) { // Loads data: off-chip --> local (matrix A and B) GlobalToLocalDirectA(agm, alm, a_ld, a_offset, tid, kwg, a_transpose, a_conjugate); @@ -351,7 +345,7 @@ __kernel void XgemmDirect(const int kSizeM, const int kSizeN, const int kSizeK, barrier(CLK_LOCAL_MEM_FENCE); // Loops over all workitem tiles, unrolled by a factor KWID - for (int pwi=0; pwi<KWGD; pwi+=KWID) { + for (int pwi=0; pwi<WGD; pwi+=KWID) { #pragma unroll for (int pit=0; pit<KWID; ++pit) { int kg = pwi + pit; |