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

[PYTHON][KERNEL] Better handling of case where cache directory already

Browse Source 
exists
This commit is contained in:
Philippe Tillet
2020-04-07 13:18:11 -04:00
committed by Philippe Tillet
parent 5943baa53f
commit 94e8ee7f01
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
python/triton/kernel.py
View File

@@ -123,8 +123,10 @@ def _make_framework_op(arg_types):
# path of .cpp and .so file # path of .cpp and .so file
home = os.path.expanduser('~') home = os.path.expanduser('~')
root = os.path.join(home, '.triton', 'torch', name) root = os.path.join(home, '.triton', 'torch', name)
if not os.path.exists(root): try:
os.makedirs(root) os.makedirs(root)
except FileExistsError:
pass
suffix = sysconfig.get_config_var('EXT_SUFFIX') suffix = sysconfig.get_config_var('EXT_SUFFIX')
so = os.path.join(root, f'op{suffix}') so = os.path.join(root, f'op{suffix}')
cpp = os.path.join(root, f'op.cpp') cpp = os.path.join(root, f'op.cpp')