未验证 提交 c94a4b94 编写于 作者: Z Zhou Wei 提交者: GitHub

Separate AVX and NO_AVX compilation, enhance installation error message (#30413)

上级 6e0da01c
...@@ -15,7 +15,7 @@ else() ...@@ -15,7 +15,7 @@ else()
set(paddle_known_gpu_archs7 "30 35 50 52") set(paddle_known_gpu_archs7 "30 35 50 52")
set(paddle_known_gpu_archs8 "30 35 50 52 60 61") set(paddle_known_gpu_archs8 "30 35 50 52 60 61")
set(paddle_known_gpu_archs9 "30 35 50 52 60 61 70") set(paddle_known_gpu_archs9 "30 35 50 52 60 61 70")
set(paddle_known_gpu_archs10 "30 35 50 52 60 61 70 75") set(paddle_known_gpu_archs10 "35 50 52 60 61 70 75")
set(paddle_known_gpu_archs11 "52 60 61 70 75 80") set(paddle_known_gpu_archs11 "52 60 61 70 75 80")
endif() endif()
......
...@@ -283,16 +283,24 @@ if avx_supported(): ...@@ -283,16 +283,24 @@ if avx_supported():
from .core_avx import _remove_tensor_list_mmap_fds from .core_avx import _remove_tensor_list_mmap_fds
except Exception as e: except Exception as e:
if has_avx_core: if has_avx_core:
sys.stderr.write(
'Error: Can not import avx core while this file exists: ' +
current_path + os.sep + 'core_avx.' + core_suffix + '\n')
raise e raise e
else: else:
from .. import compat as cpt from .. import compat as cpt
sys.stderr.write( sys.stderr.write(
'WARNING: Do not have avx core. You may not build with AVX, ' "WARNING: AVX is supported on local machine, but you have installed "
'but AVX is supported on local machine.\n You could build paddle ' "paddlepaddle without avx core. Hence, no_avx core which has worse "
'WITH_AVX=ON to get better performance.\n' "preformance will be imported.\nYou could reinstall paddlepaddle by "
'The original error is: %s\n' % cpt.get_exception_message(e)) "'python -m pip install -U paddlepaddle-gpu[==version]' or rebuild "
"paddlepaddle WITH_AVX=ON to get better performance.\n"
"The original error is: %s\n" % cpt.get_exception_message(e))
load_noavx = True load_noavx = True
else: else:
sys.stderr.write(
"WARNING: AVX is not support on your machine. Hence, no_avx core will be imported, "
"It has much worse preformance than avx core.\n")
load_noavx = True load_noavx = True
if load_noavx: if load_noavx:
...@@ -330,8 +338,14 @@ if load_noavx: ...@@ -330,8 +338,14 @@ if load_noavx:
except Exception as e: except Exception as e:
if has_noavx_core: if has_noavx_core:
sys.stderr.write( sys.stderr.write(
'Error: Can not import noavx core while this file exists ' + 'Error: Can not import noavx core while this file exists: ' +
current_path + os.sep + 'core_noavx.' + core_suffix + '\n') current_path + os.sep + 'core_noavx.' + core_suffix + '\n')
else:
sys.stderr.write(
"Error: AVX is not support on your machine, but you have installed "
"paddlepaddle with avx core, you should reinstall paddlepaddle by "
"'python -m pip install -U paddlepaddle-gpu[==version] -f "
"https://paddlepaddle.org.cn/whl/stable_noavx.html'\n")
raise e raise e
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册