1 2 #ifndef USERGEMM_CL_KERNELS_H 3 #define USERGEMM_CL_KERNELS_H 4 5 #if defined( __APPLE__ ) || defined( __MACOSX ) 6 #include <OpenCL/cl.h> 7 #else 8 #include <CL/cl.h> 9 #endif 10 11 extern cl_kernel sgemm_Col_NT_B1_MX128_NX128_KX16_clKernel; 12 13 extern cl_kernel sgemm_Col_NT_B1_MX032_NX064_KX16_ROW_clKernel; 14 extern cl_kernel sgemm_Col_NT_B1_MX064_NX032_KX16_COLUMN_clKernel; 15 extern cl_kernel sgemm_Col_NT_B1_MX032_NX032_KX16_SINGLE_clKernel; 16 17 extern cl_kernel sgemm_Col_NN_B1_MX032_NX032_KX16_BRANCH_clKernel; 18 extern cl_kernel sgemm_Col_NT_B1_MX032_NX032_KX16_BRANCH_clKernel; 19 extern cl_kernel sgemm_Col_TN_B1_MX032_NX032_KX16_BRANCH_clKernel; 20 21 static const int user_kernel_count = 7; 22 23 #ifdef __cplusplus 24 extern "C" { 25 #endif 26 void initUserGemmClKernels(void); 27 #ifdef __cplusplus 28 } 29 #endif 30 31 #endif 32