Skip to content
Snippets Groups Projects
Commit 0e9f33bd authored by Stephan Seitz's avatar Stephan Seitz
Browse files

Fix bug in TorchModule.compile that requires `compile_module_name` to be set

parent c109ac02
No related merge requests found
Pipeline #22351 failed with stage
in 2 minutes and 41 seconds
......@@ -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)
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment