diff options
| author | Gard Spreemann <gspr@nonempty.org> | 2023-06-08 11:52:00 +0200 |
|---|---|---|
| committer | Gard Spreemann <gspr@nonempty.org> | 2023-06-08 11:52:00 +0200 |
| commit | 63870a2e60c1bc8bfa7e3672457b551a8e51ffaf (patch) | |
| tree | fe2c0cd5f62e3fbd17e58d3903ec6bb37983f620 /src/kernels/level3/xgemm_batched.opencl | |
| parent | d31fb141cb597aaf405674621aa25f263aa375e1 (diff) | |
| parent | b0b302889cc786907efb080c4e1beea30d2fa39f (diff) | |
Merge tag '1.6.0' into gspr/post-bookworm
Diffstat (limited to 'src/kernels/level3/xgemm_batched.opencl')
| -rw-r--r-- | src/kernels/level3/xgemm_batched.opencl | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/src/kernels/level3/xgemm_batched.opencl b/src/kernels/level3/xgemm_batched.opencl index b51e6298..41d07d19 100644 --- a/src/kernels/level3/xgemm_batched.opencl +++ b/src/kernels/level3/xgemm_batched.opencl @@ -19,7 +19,11 @@ R"( // ================================================================================================= #if defined(ROUTINE_GEMMBATCHED) -__kernel __attribute__((reqd_work_group_size(MDIMC, NDIMC, 1))) +#if RELAX_WORKGROUP_SIZE == 1 + __kernel +#else + __kernel __attribute__((reqd_work_group_size(MDIMC, NDIMC, 1))) +#endif void XgemmBatched(const int kSizeM, const int kSizeN, const int kSizeK, const __constant real_arg* arg_alphas, const __constant real_arg* arg_betas, @@ -62,7 +66,11 @@ void XgemmBatched(const int kSizeM, const int kSizeN, const int kSizeK, // ================================================================================================= #if defined(ROUTINE_GEMMSTRIDEDBATCHED) -__kernel __attribute__((reqd_work_group_size(MDIMC, NDIMC, 1))) +#if RELAX_WORKGROUP_SIZE == 1 + __kernel +#else + __kernel __attribute__((reqd_work_group_size(MDIMC, NDIMC, 1))) +#endif void XgemmStridedBatched(const int kSizeM, const int kSizeN, const int kSizeK, const real_arg arg_alpha, const real_arg arg_beta, const __global realM* restrict agm, const int a_one, const int a_two, |