diff options
author | Cedric Nugteren <web@cedricnugteren.nl> | 2018-05-13 21:01:46 +0200 |
---|---|---|
committer | Cedric Nugteren <web@cedricnugteren.nl> | 2018-05-13 21:01:46 +0200 |
commit | ad8f1027abf6fecbf8f119172644e729a8a94d0c (patch) | |
tree | eb7e6e77321e6b8b05424a5bf1ce6693cde48852 /src/kernels | |
parent | 4e6d30088d7c73f13b0ad6db5794b232add2b735 (diff) |
Plugged in the code of strided-batched-gemm into convgemm in preparation of a new kernel
Diffstat (limited to 'src/kernels')
-rw-r--r-- | src/kernels/level3/xgemm_direct_batched.opencl | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/kernels/level3/xgemm_direct_batched.opencl b/src/kernels/level3/xgemm_direct_batched.opencl index d15ed31e..aebbb5ad 100644 --- a/src/kernels/level3/xgemm_direct_batched.opencl +++ b/src/kernels/level3/xgemm_direct_batched.opencl @@ -105,7 +105,7 @@ void XgemmDirectBatchedTT(const int kSizeM, const int kSizeN, const int kSizeK, #endif // ================================================================================================= -#if defined(ROUTINE_GEMMSTRIDEDBATCHED) +#if defined(ROUTINE_GEMMSTRIDEDBATCHED) || defined(ROUTINE_CONVGEMM) // Direct version of the strided-batched GEMM kernel with [A, B] = [non-transposed, non-transposed] __kernel __attribute__((reqd_work_group_size(MDIMCD, NDIMCD, 1))) |