From 2939e580350d648117575daa276021b3468aea69 Mon Sep 17 00:00:00 2001 From: Stephan Seitz <stephan.seitz@fau.de> Date: Thu, 2 Jan 2020 13:02:26 +0100 Subject: [PATCH] Use get_compiler_config()['command'] as host compiler for Torch Modules --- src/pystencils_autodiff/backends/astnodes.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/pystencils_autodiff/backends/astnodes.py b/src/pystencils_autodiff/backends/astnodes.py index 286b39c..95627c5 100644 --- a/src/pystencils_autodiff/backends/astnodes.py +++ b/src/pystencils_autodiff/backends/astnodes.py @@ -119,6 +119,9 @@ class TorchModule(JinjaCppFile): if not exists(file_name): write_file(file_name, source_code) + # Torch regards CXX + os.environ['CXX'] = get_compiler_config()['command'] + torch_extension = load(hash, [file_name], with_cuda=self.is_cuda, -- GitLab