Merge pull request #59 from bmerry/fix-54

Use appdirs package to create cache in XDG-correct location
inducer · Nov 7, 2014 · 7edecc1 · 7edecc1
2 parents 9a86c76 + dbbb6fe
commit 7edecc1
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/pycuda/compiler.py b/pycuda/compiler.py
@@ -208,13 +208,13 @@ def compile(source, nvcc="nvcc", options=None, keep=False,
 
     if cache_dir is None:
         from os.path import join
-        from tempfile import gettempdir
-        cache_dir = join(gettempdir(),
-                "pycuda-compiler-cache-v1-%s" % _get_per_user_string())
+        import appdirs
+        cache_dir = os.path.join(appdirs.user_cache_dir("pycuda", "pycuda"),
+                "compiler-cache-v1")
 
-        from os import mkdir
+        from os import makedirs
         try:
-            mkdir(cache_dir)
+            makedirs(cache_dir)
         except OSError, e:
             from errno import EEXIST
             if e.errno != EEXIST:

diff --git a/setup.py b/setup.py
@@ -179,7 +179,8 @@ def main():
             install_requires=[
                 "pytools>=2011.2",
                 "pytest>=2",
-                "decorator>=3.2.0"
+                "decorator>=3.2.0",
+                "appdirs>=1.4.0"
                 ],
 
             ext_package="pycuda",