Adapt interface to KMM 0.3

NLeSC-COMPAS · Nov 27, 2024 · 50d8811 · 50d8811
1 parent ef23299
commit 50d8811
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/trajectories/signal.cu b/src/trajectories/signal.cu
@@ -131,13 +131,13 @@ void magnetization_to_signal_cartesian_gemm(
         cuComplex alpha = {1, 0};
         cuComplex beta = {0, 0};
 
-        gpuDataType_t output_type = GPU_C_32F;
-        gpuDataType_t input_type = GPU_C_32F;
+        cudaDataType_t output_type = GPU_C_32F;
+        cudaDataType_t input_type = GPU_C_32F;
         cublasGemmAlgo_t compute_algo = CUBLAS_GEMM_DEFAULT;
 
         COMPAS_GPU_CHECK(cublasSetStream(context.cublas(), nullptr));
         COMPAS_GPU_CHECK(cublasGemmEx(
-            context.cublas(),
+            context.blas(),
             CUBLAS_OP_T,  // transa
             CUBLAS_OP_N,  // transb
             samples_per_readout,  // m