gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[PYTHON] move ephemeral files to homedir (#549)

Browse Source 
This prevents potential conflicts with other users on shared machines.
This commit is contained in:
Madeleine Thompson
2022-06-13 19:37:52 -07:00
committed by GitHub
parent 93209c07e0
commit 8ce2c12e33
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
python/setup.py
View File

@@ -27,10 +27,11 @@ def get_llvm():
return '', '' return '', ''
# download if nothing is installed # download if nothing is installed
name = 'clang+llvm-11.0.1-x86_64-linux-gnu-ubuntu-16.04' name = 'clang+llvm-11.0.1-x86_64-linux-gnu-ubuntu-16.04'
dir = '/tmp' dir = os.path.join(os.environ["HOME"], ".triton", "llvm")
llvm_include_dir = '{dir}/{name}/include'.format(dir=dir, name=name) llvm_include_dir = '{dir}/{name}/include'.format(dir=dir, name=name)
llvm_library_dir = '{dir}/{name}/lib'.format(dir=dir, name=name) llvm_library_dir = '{dir}/{name}/lib'.format(dir=dir, name=name)
if not os.path.exists(llvm_library_dir): if not os.path.exists(llvm_library_dir):
os.makedirs(dir, exist_ok=True)
try: try:
shutil.rmtree(os.path.join(dir, name)) shutil.rmtree(os.path.join(dir, name))
except Exception: except Exception:

3
python/triton/code_gen.py
View File

@@ -1158,8 +1158,7 @@ class DependenciesFinder(ast.NodeVisitor):
def default_cache_dir(): def default_cache_dir():
import getpass return os.path.join(os.environ["HOME"], ".triton", "cache")
return f'/tmp/triton_{getpass.getuser()}'
class JITFunction: class JITFunction: