Skip to content

Commit

Permalink
Guard hipblas path change with a condition on HIP_VERSION.
Browse files Browse the repository at this point in the history
  • Loading branch information
nkoukpaizan committed Jan 9, 2025
1 parent 84bb8cf commit f372155
Show file tree
Hide file tree
Showing 3 changed files with 12 additions and 0 deletions.
4 changes: 4 additions & 0 deletions src/LinAlg/VectorHipKernels.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -53,7 +53,11 @@
*/
#include "VectorHipKernels.hpp"
#include <hip/hip_runtime.h>
#if HIP_VERSION >= 50200000
#include <hipblas/hipblas.h>
#else
#include <hipblas.h>
#endif

#include <thrust/device_ptr.h>
#include <thrust/device_vector.h>
Expand Down
4 changes: 4 additions & 0 deletions src/LinAlg/hiopVectorHip.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -59,7 +59,11 @@
#include "VectorHipKernels.hpp"
#include "MathKernelsHip.hpp"
#include <hip/hip_runtime.h>
#if HIP_VERSION >= 50200000
#include <hipblas/hipblas.h>
#else
#include <hipblas.h>
#endif

#include "hiopVectorPar.hpp"

Expand Down
4 changes: 4 additions & 0 deletions src/LinAlg/hiopVectorHip.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -65,7 +65,11 @@

#include "hiopVector.hpp"

#if HIP_VERSION >= 50200000
#include <hipblas/hipblas.h>
#else
#include <hipblas.h>
#endif

namespace hiop
{
Expand Down

0 comments on commit f372155

Please sign in to comment.