Made the identification of temporary files directory more robust

16f43ede · Peter Eastman · bd22eada · 16f43ede
Commit 16f43ede authored Jun 20, 2012 by Peter Eastman
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletion

platforms/cuda2/src/CudaPlatform.cpp platforms/cuda2/src/CudaPlatform.cpp +5 -1

No files found.
--- a/platforms/cuda2/src/CudaPlatform.cpp
+++ b/platforms/cuda2/src/CudaPlatform.cpp
@@ -35,6 +35,7 @@
 #include <algorithm>
 #include <cctype>
 #include <sstream>
+#include <cstdio>

 using namespace OpenMM;
 using namespace std;
@@ -87,7 +88,10 @@ CudaPlatform::CudaPlatform() {
    setPropertyDefaultValue(CudaTempDirectory(), string(getenv("TEMP")));
 #else
    setPropertyDefaultValue(CudaCompiler(), "/usr/local/cuda/bin/nvcc");
-    setPropertyDefaultValue(CudaTempDirectory(), string(getenv("TMPDIR")));
+    char* tmpdir = getenv("TMPDIR");
+    if (tmpdir == NULL)
+        tmpdir = P_tmpdir;
+    setPropertyDefaultValue(CudaTempDirectory(), string(tmpdir));
 #endif
 }