提交 649d40c6 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Tag compilers that support/don't support amdlibm.

上级 ec3cef2e
...@@ -1770,6 +1770,8 @@ class GCC_compiler(Compiler): ...@@ -1770,6 +1770,8 @@ class GCC_compiler(Compiler):
# The equivalent flags of --march=native used by g++. # The equivalent flags of --march=native used by g++.
march_flags = None march_flags = None
supports_amdlibm = True
@staticmethod @staticmethod
def version_str(): def version_str():
return theano.config.cxx + " " + gcc_version_str return theano.config.cxx + " " + gcc_version_str
......
...@@ -131,6 +131,8 @@ def add_standard_rpath(rpath): ...@@ -131,6 +131,8 @@ def add_standard_rpath(rpath):
class NVCC_compiler(Compiler): class NVCC_compiler(Compiler):
supports_amdlibm = False
@staticmethod @staticmethod
def try_compile_tmp(src_code, tmp_prefix='', flags=(), def try_compile_tmp(src_code, tmp_prefix='', flags=(),
try_run=False, output=False): try_run=False, output=False):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论