From fbb484f89774383ec88b165e3e20834a1c6b120e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Raffaele=20Solc=C3=A0?= <rasolca@cscs.ch>
Date: Fri, 22 Nov 2024 15:43:42 +0100
Subject: [PATCH] fix headers

---
 include/dlaf/gpu/blas/gpublas.h | 2 ++
 src/lapack/gpu/larft.cu         | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/include/dlaf/gpu/blas/gpublas.h b/include/dlaf/gpu/blas/gpublas.h
index 8fb3e07832..45aa734c85 100644
--- a/include/dlaf/gpu/blas/gpublas.h
+++ b/include/dlaf/gpu/blas/gpublas.h
@@ -12,6 +12,7 @@
 /// @file
 /// Provides gpublas wrappers for BLAS operations.
 
+#ifdef DLAF_WITH_GPU
 #include <cstddef>
 #include <utility>
 
@@ -139,3 +140,4 @@ DLAF_MAKE_GPUBLAS_OP(Trmm, trmm_outofplace);
 
 DLAF_MAKE_GPUBLAS_OP(Trsm, trsm);
 }
+#endif
diff --git a/src/lapack/gpu/larft.cu b/src/lapack/gpu/larft.cu
index fc0c887e5e..3c8078e626 100644
--- a/src/lapack/gpu/larft.cu
+++ b/src/lapack/gpu/larft.cu
@@ -8,11 +8,11 @@
 // SPDX-License-Identifier: BSD-3-Clause
 //
 
-#include <cublas_v2.h>
 #include <whip.hpp>
 
 #include <dlaf/common/assert.h>
 #include <dlaf/gpu/assert.cu.h>
+#include <dlaf/gpu/blas/api.h>
 #include <dlaf/gpu/blas/gpublas.h>
 #include <dlaf/lapack/gpu/larft.h>
 #include <dlaf/types.h>