From 7fa8d62a790a9716b745f28a8c99a12873e5c1be Mon Sep 17 00:00:00 2001 From: Ryker Fish Date: Wed, 20 Aug 2025 22:54:29 -0600 Subject: [PATCH] fix cublas defs in lanczos to respect double precision --- include/lanczos/utils/device_blas.h | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/include/lanczos/utils/device_blas.h b/include/lanczos/utils/device_blas.h index 5fc3e93..cb893b1 100644 --- a/include/lanczos/utils/device_blas.h +++ b/include/lanczos/utils/device_blas.h @@ -1,7 +1,6 @@ #pragma once #include "cublasDebug.h" #include "cuda_lib_defines.h" -#include "defines.h" namespace lanczos { struct Blas { cublasHandle_t cublas_handle; @@ -9,23 +8,23 @@ struct Blas { ~Blas() { CublasSafeCall(cublasDestroy(cublas_handle)); } template void gemv(Args &&...args) { CublasSafeCall( - cublasSgemv(cublas_handle, CUBLAS_OP_N, std::forward(args)...)); + cublasgemv(cublas_handle, CUBLAS_OP_N, std::forward(args)...)); } template void nrm2(Args &&...args) { - CublasSafeCall(cublasSnrm2(cublas_handle, std::forward(args)...)); + CublasSafeCall(cublasnrm2(cublas_handle, std::forward(args)...)); } template void axpy(Args &&...args) { - CublasSafeCall(cublasSaxpy(cublas_handle, std::forward(args)...)); + CublasSafeCall(cublasaxpy(cublas_handle, std::forward(args)...)); } template void dot(Args &&...args) { - CublasSafeCall(cublasSdot(cublas_handle, std::forward(args)...)); + CublasSafeCall(cublasdot(cublas_handle, std::forward(args)...)); } template void scal(Args &&...args) { - CublasSafeCall(cublasSscal(cublas_handle, std::forward(args)...)); + CublasSafeCall(cublasscal(cublas_handle, std::forward(args)...)); } }; } // namespace lanczos