Fixed potential compilation error in CUDA FFT

3fd5e2c5 · Peter Eastman · 363406d1 · 3fd5e2c5
Commit 3fd5e2c5 authored Jul 24, 2015 by Peter Eastman
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

platforms/cuda/src/CudaFFT3D.cpp platforms/cuda/src/CudaFFT3D.cpp +1 -1

No files found.
--- a/platforms/cuda/src/CudaFFT3D.cpp
+++ b/platforms/cuda/src/CudaFFT3D.cpp
@@ -169,7 +169,7 @@ static int getSmallestRadix(int size) {
 }
 CUfunction CudaFFT3D::createKernel(int xsize, int ysize, int zsize, int& threads, int axis, bool forward, bool inputIsReal) {
-    int maxThreads = 256;
+    int maxThreads = (context.getUseDoublePrecision() ? 128 : 256);
 //    while (maxThreads > 128 && maxThreads-64 >= zsize)
 //        maxThreads -= 64;
    int threadsPerBlock = zsize/getSmallestRadix(zsize);