portelli/Grid
1
0
Fork 0
mirror of https://github.com/paboyle/Grid.git synced 2024-11-09 23:45:36 +00:00
Code Releases Activity

Blas fix

Browse Source
This commit is contained in:
Peter Boyle 2024-08-27 19:53:09 +00:00
parent 4e2a6d87c4
commit af9ea0864c
1 changed files with 12 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
BLAS_benchmark/BatchBlasBench.cc
View File

@ -382,14 +382,14 @@ public:
if ( OpB == GridBLAS_OP_C ) iOpB = oneapi::mkl::transpose::C; if ( OpB == GridBLAS_OP_C ) iOpB = oneapi::mkl::transpose::C;
oneapi::mkl::blas::column_major::gemm(*gridblasHandle, oneapi::mkl::blas::column_major::gemm(*gridblasHandle,
&iOpA, iOpA,
&iOpB, iOpB,
&m64,&n64,&k64, m64,n64,k64,
(ComplexD *) &alpha_p[0], (ComplexD *) &alpha_p[0],
(const ComplexD *)Amk, (const int64_t *)&lda64, (const ComplexD *)Amk, (int64_t )lda64,
(const ComplexD *)Bkn, (const int64_t *)&ldb64, (const ComplexD *)Bkn, (int64_t )ldb64,
(ComplexD *) &beta_p[0], (ComplexD *) &beta_p[0],
(ComplexD *)Cmn, (const int64_t *)&ldc64); (ComplexD *)Cmn, (int64_t)ldc64);
synchronise(); synchronise();
#endif #endif
#if !defined(GRID_SYCL) && !defined(GRID_CUDA) && !defined(GRID_HIP) #if !defined(GRID_SYCL) && !defined(GRID_CUDA) && !defined(GRID_HIP)
@ -510,14 +510,14 @@ public:
if ( OpB == GridBLAS_OP_C ) iOpB = oneapi::mkl::transpose::C; if ( OpB == GridBLAS_OP_C ) iOpB = oneapi::mkl::transpose::C;
oneapi::mkl::blas::column_major::gemm(*gridblasHandle, oneapi::mkl::blas::column_major::gemm(*gridblasHandle,
&iOpA, iOpA,
&iOpB, iOpB,
&m64,&n64,&k64, m64,n64,k64,
(ComplexF *) &alpha_p[0], (ComplexF *) &alpha_p[0],
(const ComplexF *)Amk, (const int64_t *)&lda64, (const ComplexF *)Amk, (int64_t )lda64,
(const ComplexF *)Bkn, (const int64_t *)&ldb64, (const ComplexF *)Bkn, (int64_t )ldb64,
(ComplexF *) &beta_p[0], (ComplexF *) &beta_p[0],
(ComplexF *)Cmn, (const int64_t *)&ldc64); (ComplexF *)Cmn, (int64_t )ldc64);
synchronise(); synchronise();
#endif #endif
#if !defined(GRID_SYCL) && !defined(GRID_CUDA) && !defined(GRID_HIP) #if !defined(GRID_SYCL) && !defined(GRID_CUDA) && !defined(GRID_HIP)