Fixed bad assumption about null pointer return from cudaFree which ca…

…used a segmentation fault. Closes #37.
lattice · Oct 18, 2011 · 981e1bf · 981e1bf
1 parent 3de6e8f
commit 981e1bf
Showing 1 changed file with 24 additions and 6 deletions.
diff --git a/lib/blas_quda.cu b/lib/blas_quda.cu
@@ -151,12 +151,30 @@ void initBlas(void)
 
 void endBlas(void)
 {
-  if (d_reduceFloat) cudaFree(d_reduceFloat);
-  if (d_reduceComplex) cudaFree(d_reduceComplex);
-  if (d_reduceFloat3) cudaFree(d_reduceFloat3);
-  if (h_reduceFloat) cudaFreeHost(h_reduceFloat);
-  if (h_reduceComplex) cudaFreeHost(h_reduceComplex);
-  if (h_reduceFloat3) cudaFreeHost(h_reduceFloat3);
+  if (d_reduceFloat) {
+    cudaFree(d_reduceFloat);
+    d_reduceFloat = 0;
+  }
+  if (d_reduceComplex) {
+    cudaFree(d_reduceComplex);
+    d_reduceComplex = 0;
+  }
+  if (d_reduceFloat3) {
+    cudaFree(d_reduceFloat3);
+    d_reduceFloat3 = 0;
+  }
+  if (h_reduceFloat) {
+    cudaFreeHost(h_reduceFloat);
+    h_reduceFloat = 0;
+  }
+  if (h_reduceComplex) {
+    cudaFreeHost(h_reduceComplex);
+    h_reduceComplex = 0;
+  }
+  if (h_reduceFloat3) {
+    cudaFreeHost(h_reduceFloat3);
+    h_reduceFloat3 = 0;
+  }
 }
 
 void setBlasTuning(QudaTune tune)