Commit aa5f89f0 authored by Stephan Seitz's avatar Stephan Seitz
Browse files

Use pystencils compiler as torch host compiler

parent 22007af0
Pipeline #20585 passed with stage
in 3 minutes and 49 seconds
......@@ -14,7 +14,7 @@ from collections.abc import Iterable
from os.path import dirname, exists, join
from pystencils.astnodes import FieldPointerSymbol, FieldShapeSymbol, FieldStrideSymbol
from pystencils.cpu.cpujit import get_cache_config
from pystencils.cpu.cpujit import get_cache_config, get_compiler_config
from pystencils.include import get_pycuda_include_path, get_pystencils_include_path
from pystencils_autodiff._file_io import read_template_from_file, write_file
from pystencils_autodiff.backends.python_bindings import (
......@@ -123,7 +123,7 @@ class TorchModule(JinjaCppFile):
[file_name],
with_cuda=self.is_cuda,
extra_cflags=['--std=c++14'],
extra_cuda_cflags=['-std=c++14'],
extra_cuda_cflags=['-std=c++14', '--ccbin', get_compiler_config()['command']],
build_directory=build_dir,
extra_include_paths=[get_pycuda_include_path(),
get_pystencils_include_path()])
......
Markdown is supported
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