From b24e4cdddd5cb43fdad3c88e52aed45518d16bce Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Fri, 13 Sep 2019 00:54:00 +0200
Subject: [PATCH] Set get_compiler_config()['command'] to 'cl.exe' on Windows
---
src/pystencils_autodiff/tensorflow_jit.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/src/pystencils_autodiff/tensorflow_jit.py b/src/pystencils_autodiff/tensorflow_jit.py
index 1384893..4bf3a85 100644
--- a/src/pystencils_autodiff/tensorflow_jit.py
+++ b/src/pystencils_autodiff/tensorflow_jit.py
@@ -13,11 +13,11 @@ import subprocess
import sysconfig
from os.path import exists, join
+from pystencils_autodiff._file_io import read_file, write_file
from tqdm import tqdm
import pystencils
from pystencils.cpu.cpujit import get_cache_config, get_compiler_config, get_pystencils_include_path
-from pystencils_autodiff._file_io import read_file, write_file
_hash = hashlib.md5
@@ -29,6 +29,7 @@ if get_compiler_config()['os'] != 'windows':
_include_flags = ['-I' + sysconfig.get_paths()['include'], '-I' + get_pystencils_include_path()]
_do_not_link_flag = "-c"
_position_independent_flag = "-fPIC"
+ get_compiler_config()['command'] = 'cl.exe'
else:
_do_not_link_flag = "/c"
_output_flag = '/OUT:'
--
GitLab