提交 3746efc8 authored 作者: Frederic's avatar Frederic

Fix crash, the amdlibm hack to remove it for nvcc change the hash./memtestCL…

Fix crash, the amdlibm hack to remove it for nvcc change the hash./memtestCL --gpu 0 4000 10000 This also make use keep the lock for less time.
上级 c356183f
...@@ -1304,12 +1304,14 @@ class CLinker(link.Linker): ...@@ -1304,12 +1304,14 @@ class CLinker(link.Linker):
preargs.remove('-DREPLACE_WITH_AMDLIBM') preargs.remove('-DREPLACE_WITH_AMDLIBM')
if 'amdlibm' in libs: if 'amdlibm' in libs:
libs.remove('amdlibm') libs.remove('amdlibm')
# We want to compute the code without the lock
src_code = mod.code()
get_lock() get_lock()
try: try:
_logger.debug("LOCATION %s", str(location)) _logger.debug("LOCATION %s", str(location))
module = c_compiler.compile_str( module = c_compiler.compile_str(
module_name=mod.code_hash, module_name=mod.code_hash,
src_code=mod.code(), src_code=src_code,
location=location, location=location,
include_dirs=self.header_dirs(), include_dirs=self.header_dirs(),
lib_dirs=self.lib_dirs(), lib_dirs=self.lib_dirs(),
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论