Mentions légales du service

Skip to content
Snippets Groups Projects
Commit 01075844 authored by Mathieu Faverge's avatar Mathieu Faverge
Browse files

hip: Fix half definition when using hip

parent 5245b134
No related branches found
No related tags found
1 merge request!518gitlab-ci-initial-cache.cmake: CMAKE_C_FLAGS must be set as a cache variable...
......@@ -19,11 +19,11 @@
int
HIP_hgemm( cham_trans_t transa, cham_trans_t transb,
int m, int n, int k,
const CHAMELEON_Real16_t *alpha,
const CHAMELEON_Real16_t *A, int lda,
const CHAMELEON_Real16_t *B, int ldb,
const CHAMELEON_Real16_t *beta,
CHAMELEON_Real16_t *C, int ldc,
const hipblasHalf *alpha,
const hipblasHalf *A, int lda,
const hipblasHalf *B, int ldb,
const hipblasHalf *beta,
hipblasHalf *C, int ldc,
hipblasHandle_t handle )
{
hipblasStatus_t rc;
......
......@@ -62,11 +62,11 @@ BEGIN_C_DECLS
int HIP_hgemm( cham_trans_t transa, cham_trans_t transb,
int m, int n, int k,
const CHAMELEON_Real16_t *alpha,
const CHAMELEON_Real16_t *A, int lda,
const CHAMELEON_Real16_t *B, int ldb,
const CHAMELEON_Real16_t *beta,
CHAMELEON_Real16_t *C, int ldc,
const hipblasHalf *alpha,
const hipblasHalf *A, int lda,
const hipblasHalf *B, int ldb,
const hipblasHalf *beta,
hipblasHalf *C, int ldc,
hipblasHandle_t handle );
END_C_DECLS
......
......@@ -109,6 +109,8 @@ typedef int8_t cham_bool_t;
*/
#if defined(__cplusplus) && defined(CHAMELEON_USE_CUDA) && (CUDA_VERSION >= 7500)
typedef __half CHAMELEON_Real16_t;
#elif defined(CHAMELEON_USE_HIP) && defined(__HIP__)
typedef hipblasHalf CHAMELEON_Real16_t;
#else
/* use short for cuda older than 7.5 and non-cuda files
* corresponding routines would not work anyway since there is no half precision */
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment