fix headers

eth-cscs · Nov 22, 2024 · fbb484f · fbb484f
1 parent 4822555
commit fbb484f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git 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
@@ -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>