From 0e9f33bd78e1cb0ea85965abc62e998242a194cc Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Fri, 28 Feb 2020 13:06:23 +0100
Subject: [PATCH] Fix bug in TorchModule.compile that requires
`compile_module_name` to be set
---
src/pystencils_autodiff/backends/astnodes.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/pystencils_autodiff/backends/astnodes.py b/src/pystencils_autodiff/backends/astnodes.py
index 10228c8..a040b06 100644
--- a/src/pystencils_autodiff/backends/astnodes.py
+++ b/src/pystencils_autodiff/backends/astnodes.py
@@ -154,7 +154,7 @@ class TorchModule(JinjaCppFile):
os.makedirs(build_dir, exist_ok=True)
file_name = join(build_dir, f'{hash}{file_extension}')
- self.compiled_file = (join(build_dir, compile_module_name) or file_name).replace('.cpp', '') + '.so'
+ self.compiled_file = join(build_dir, compile_module_name or file_name).replace('.cpp', '') + '.so'
if not exists(file_name):
write_file(file_name, source_code)
--
GitLab