Mentions légales du service

Skip to content
Snippets Groups Projects
Commit 443d1a86 authored by Xavier Lacoste's avatar Xavier Lacoste Committed by Mathieu Faverge
Browse files

hipblas/ztrmm: Change the xtrmm API for ROCM_MAJOR_VERSION >= 6

parent af8d7852
No related branches found
No related tags found
1 merge request!455change ztrmm API for hipblas > 2.0
...@@ -11,11 +11,11 @@ ...@@ -11,11 +11,11 @@
* *
* @brief Chameleon hip_ztrmm GPU kernel * @brief Chameleon hip_ztrmm GPU kernel
* *
* @version 1.2.0 * @version 1.3.0
* @author Florent Pruvost * @author Florent Pruvost
* @author Mathieu Faverge * @author Mathieu Faverge
* @author Loris Lucido * @author Loris Lucido
* @date 2022-02-22 * @date 2024-04-03
* @precisions normal z -> c d s * @precisions normal z -> c d s
* *
*/ */
...@@ -32,15 +32,24 @@ HIP_ztrmm( cham_side_t side, cham_uplo_t uplo, ...@@ -32,15 +32,24 @@ HIP_ztrmm( cham_side_t side, cham_uplo_t uplo,
{ {
hipblasStatus_t rc; hipblasStatus_t rc;
#if (ROCM_VERSION_MAJOR < 6)
rc = hipblasZtrmm( handle, rc = hipblasZtrmm( handle,
chameleon_hipblas_const(side), chameleon_hipblas_const(uplo), chameleon_hipblas_const(side), chameleon_hipblas_const(uplo),
chameleon_hipblas_const(transa), chameleon_hipblas_const(diag), chameleon_hipblas_const(transa), chameleon_hipblas_const(diag),
m, n, m, n,
HIPBLAS_VALUE(alpha), A, lda, HIPBLAS_VALUE(alpha), A, lda,
B, ldb ); B, ldb );
#else
rc = hipblasZtrmm( handle,
chameleon_hipblas_const(side), chameleon_hipblas_const(uplo),
chameleon_hipblas_const(transa), chameleon_hipblas_const(diag),
m, n,
HIPBLAS_VALUE(alpha), A, lda,
B, ldb,
B, ldb );
#endif
assert( rc == HIPBLAS_STATUS_SUCCESS ); assert( rc == HIPBLAS_STATUS_SUCCESS );
(void)rc; (void)rc;
return CHAMELEON_SUCCESS; return CHAMELEON_SUCCESS;
} }
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment