diff options
Diffstat (limited to 'include/clblast_cuda.h')
| -rw-r--r-- | include/clblast_cuda.h | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/include/clblast_cuda.h b/include/clblast_cuda.h index e28f68e5..0f510981 100644 --- a/include/clblast_cuda.h +++ b/include/clblast_cuda.h @@ -492,7 +492,8 @@ StatusCode Gemm(const Layout layout, const Transpose a_transpose, const Transpos const CUdeviceptr b_buffer, const size_t b_offset, const size_t b_ld, const T beta, CUdeviceptr c_buffer, const size_t c_offset, const size_t c_ld, - const CUcontext context, const CUdevice device); + const CUcontext context, const CUdevice device, + CUdeviceptr temp_buffer = nullptr); // Symmetric matrix-matrix multiplication: SSYMM/DSYMM/CSYMM/ZSYMM/HSYMM template <typename T> |