From 18f91a4f61ac7cecf003fedf85371df5876155c1 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Tue, 5 Nov 2019 19:04:39 +0100
Subject: [PATCH] Replace module name in PyBind Module with hash
---
src/pystencils_autodiff/backends/astnodes.py | 12 ++++++++----
1 file changed, 8 insertions(+), 4 deletions(-)
diff --git a/src/pystencils_autodiff/backends/astnodes.py b/src/pystencils_autodiff/backends/astnodes.py
index 7a7686a..a3b1fa0 100644
--- a/src/pystencils_autodiff/backends/astnodes.py
+++ b/src/pystencils_autodiff/backends/astnodes.py
@@ -168,18 +168,22 @@ setup_pybind11(cfg)
source_code = self.CPP_IMPORT_PREFIX + str(self)
hash_str = _hash(source_code.encode()).hexdigest()
- cache_dir = join(get_cache_config()['object_cache'], f'cppimport_{hash_str}')
- file_name = join(cache_dir, f'{self.module_name}.cpp')
+ source_code_with_hash = source_code.replace(
+ f'PYBIND11_MODULE({self.module_name}',
+ f'PYBIND11_MODULE(cppimport_{hash_str}')
+
+ cache_dir = join(get_cache_config()['object_cache'])
+ file_name = join(cache_dir, f'cppimport_{hash_str}.cpp')
os.makedirs(cache_dir, exist_ok=True)
if not exists(file_name):
- write_file(file_name, source_code)
+ write_file(file_name, source_code_with_hash)
# TODO: propagate extra headers
if cache_dir not in sys.path:
sys.path.append(cache_dir)
try:
- torch_extension = cppimport.imp(self.module_name)
+ torch_extension = cppimport.imp(f'cppimport_{hash_str}')
except Exception as e:
print(e)
torch_extension = load(self.module_name, [file_name])
--
GitLab