cuda: try simplifying cufinufft signatures

janden · Dec 18, 2023 · 7826e9a · 7826e9a
1 parent 78d03ab
commit 7826e9a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/include/cufinufft.h b/include/cufinufft.h
@@ -1,7 +1,7 @@
 // Defines the C++/C user interface to CUFINUFFT library.
-#include <cufft.h>
 
 #include <stdint.h>
+#include <complex.h>
 
 #include <cufinufft_opts.h>
 #include <finufft_errors.h>
@@ -24,8 +24,8 @@ int cufinufft_setpts(cufinufft_plan d_plan, int M, double *h_kx, double *h_ky, d
 int cufinufftf_setpts(cufinufftf_plan d_plan, int M, float *h_kx, float *h_ky, float *h_kz, int N, float *h_s,
                       float *h_t, float *h_u);
 
-int cufinufft_execute(cufinufft_plan d_plan, cuDoubleComplex *h_c, cuDoubleComplex *h_fk);
-int cufinufftf_execute(cufinufftf_plan d_plan, cuFloatComplex *h_c, cuFloatComplex *h_fk);
+int cufinufft_execute(cufinufft_plan d_plan, double complex *h_c, double complex *h_fk);
+int cufinufftf_execute(cufinufftf_plan d_plan, float complex *h_c, float complex *h_fk);
 
 int cufinufft_destroy(cufinufft_plan d_plan);
 int cufinufftf_destroy(cufinufftf_plan d_plan);

diff --git a/src/cuda/cufinufft.cu b/src/cuda/cufinufft.cu
@@ -63,11 +63,11 @@ int cufinufft_setpts(cufinufft_plan d_plan, int M, double *d_kx, double *d_ky, d
     return cufinufft_setpts_impl(M, d_kx, d_ky, d_kz, N, d_s, d_t, d_u, (cufinufft_plan_t<double> *)d_plan);
 }
 
-int cufinufftf_execute(cufinufftf_plan d_plan, cuFloatComplex *d_c, cuFloatComplex *d_fk) {
+int cufinufftf_execute(cufinufftf_plan d_plan, float complex *d_c, float complex *d_fk) {
     return cufinufft_execute_impl<float>(d_c, d_fk, (cufinufft_plan_t<float> *)d_plan);
 }
 
-int cufinufft_execute(cufinufft_plan d_plan, cuDoubleComplex *d_c, cuda_complex<double> *d_fk) {
+int cufinufft_execute(cufinufft_plan d_plan, double complex *d_c, double complex *d_fk) {
     return cufinufft_execute_impl<double>(d_c, d_fk, (cufinufft_plan_t<double> *)d_plan);
 }