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>