Skip to content

Commit

Permalink
Fix building with hip but without hipblas
Browse files Browse the repository at this point in the history
starpu_hipblas_get_local_handle() will not be useful in that case
anyway. starpu_hipblas_init/shutdown being available is however
convenient.
  • Loading branch information
sthibaul committed Feb 4, 2025
1 parent 268647c commit 1535399
Showing 1 changed file with 8 additions and 4 deletions.
12 changes: 8 additions & 4 deletions include/starpu_hipblas.h
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,12 @@
#ifndef __STARPU_HIPBLAS_H__
#define __STARPU_HIPBLAS_H__

#ifdef STARPU_USE_HIP
#ifdef STARPU_USE_HIPBLAS
#include <hipblas/hipblas.h>
#endif
#endif

#ifdef __cplusplus
extern "C" {
#endif
Expand All @@ -36,17 +42,15 @@ extern "C" {
void starpu_hipblas_init(void);

#ifdef STARPU_USE_HIP
#ifndef STARPU_USE_HIPBLAS
typedef void* hipblasHandle_t;
#endif

#ifdef STARPU_USE_HIPBLAS
/**
Return the HIPBLAS handle to be used to queue HIPBLAS kernels. It
is properly initialized and configured for multistream by
starpu_hipblas_init().
*/
hipblasHandle_t starpu_hipblas_get_local_handle(void);
#endif
#endif

/**
Synchronously deinitialize the HIPBLAS library on
Expand Down

0 comments on commit 1535399

Please sign in to comment.