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

Fixup: use correct object_cache path (objectcache instead of object_cache)

parent 8a3dab1f
Branches
Tags
No related merge requests found
Pipeline #18958 failed
...@@ -108,7 +108,7 @@ class TorchModule(JinjaCppFile): ...@@ -108,7 +108,7 @@ class TorchModule(JinjaCppFile):
[file_name], [file_name],
with_cuda=self.is_cuda, with_cuda=self.is_cuda,
extra_cflags=['--std=c++14'], extra_cflags=['--std=c++14'],
build_directory=join(cache_dir, 'object_cache'), build_directory=get_cache_config()['object_cache'],
extra_include_paths=[get_pycuda_include_path(), extra_include_paths=[get_pycuda_include_path(),
get_pystencils_include_path()]) get_pystencils_include_path()])
return torch_extension return torch_extension
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment