Sfoglia il codice sorgente

Fix build without cuda

Samuel Thibault 13 anni fa
parent
commit
f9ee57e914
2 ha cambiato i file con 4 aggiunte e 0 eliminazioni
  1. 2 0
      examples/starpufft/starpufftx1d.c
  2. 2 0
      examples/starpufft/starpufftx2d.c

+ 2 - 0
examples/starpufft/starpufftx1d.c

@@ -391,6 +391,7 @@ static struct starpu_codelet STARPUFFT(twist3_1d_codelet) = {
  *
  */
 
+#ifdef STARPU_USE_CUDA
 /* Perform one fft of size n */
 static void
 STARPUFFT(fft_1d_plan_gpu)(void *args)
@@ -424,6 +425,7 @@ STARPUFFT(fft_1d_kernel_gpu)(void *descr[], void *args)
 
 	cudaStreamSynchronize(starpu_cuda_get_local_stream());
 }
+#endif
 
 #ifdef STARPU_HAVE_FFTW
 /* Perform one fft of size n */

+ 2 - 0
examples/starpufft/starpufftx2d.c

@@ -376,6 +376,7 @@ static struct starpu_codelet STARPUFFT(twist3_2d_codelet) = {
  *
  */
 
+#ifdef STARPU_USE_CUDA
 /* Perform one fft of size n,m */
 static void
 STARPUFFT(fft_2d_plan_gpu)(void *args)
@@ -410,6 +411,7 @@ STARPUFFT(fft_2d_kernel_gpu)(void *descr[], void *args)
 
 	cudaStreamSynchronize(starpu_cuda_get_local_stream());
 }
+#endif
 
 #ifdef STARPU_HAVE_FFTW
 /* Perform one fft of size n,m */