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