From 94e8ee7f01b6acac0405e5136a737042442139ba Mon Sep 17 00:00:00 2001 From: Philippe Tillet Date: Tue, 7 Apr 2020 13:18:11 -0400 Subject: [PATCH] [PYTHON][KERNEL] Better handling of case where cache directory already exists --- python/triton/kernel.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/python/triton/kernel.py b/python/triton/kernel.py index 8b6011337..5c1a1ccda 100644 --- a/python/triton/kernel.py +++ b/python/triton/kernel.py @@ -123,8 +123,10 @@ def _make_framework_op(arg_types): # path of .cpp and .so file home = os.path.expanduser('~') root = os.path.join(home, '.triton', 'torch', name) - if not os.path.exists(root): + try: os.makedirs(root) + except FileExistsError: + pass suffix = sysconfig.get_config_var('EXT_SUFFIX') so = os.path.join(root, f'op{suffix}') cpp = os.path.join(root, f'op.cpp')