提交 4a880f03 编写于 作者: Y Yu Yang

Fix bug in NO_AVX when using lstm

* libpaddle_cuda add wrong -mavx to some cpp files.
上级 3304de7a
...@@ -40,3 +40,4 @@ HPPL_ERROR_LOG ...@@ -40,3 +40,4 @@ HPPL_ERROR_LOG
unittest.list unittest.list
proto proto
dist dist
setup.py
set(AVX_SOURCES
src/hl_math.cc
src/hl_avx_functions.cc
)
set(CUDA_SOURCES set(CUDA_SOURCES
src/hl_time.cc src/hl_time.cc
src/hl_math.cc
src/hl_cpu_functions.cc src/hl_cpu_functions.cc
src/hl_avx_functions.cc) ${AVX_SOURCES})
set(CUDA_CXX_WITH_GPU_SOURCES set(CUDA_CXX_WITH_GPU_SOURCES
src/hl_cuda_cublas.cc src/hl_cuda_cublas.cc
...@@ -12,7 +15,7 @@ set(CUDA_CXX_WITH_GPU_SOURCES ...@@ -12,7 +15,7 @@ set(CUDA_CXX_WITH_GPU_SOURCES
set_source_files_properties(${CUDA_CXX_WITH_GPU_SOURCES} set_source_files_properties(${CUDA_CXX_WITH_GPU_SOURCES}
PROPERTIES COMPILE_FLAGS "-D__NVCC__") PROPERTIES COMPILE_FLAGS "-D__NVCC__")
set_source_files_properties(${CUDA_SOURCES} set_source_files_properties(${AVX_SOURCES}
PROPERTIES COMPILE_FLAGS "-mavx") PROPERTIES COMPILE_FLAGS "-mavx")
set(CUDA_DSO_SOURCES set(CUDA_DSO_SOURCES
...@@ -73,4 +76,3 @@ endif() ...@@ -73,4 +76,3 @@ endif()
add_style_check_target(paddle_cuda ${CUDA_SOURCES}) add_style_check_target(paddle_cuda ${CUDA_SOURCES})
add_style_check_target(paddle_cuda ${CUDA_HEADERS}) add_style_check_target(paddle_cuda ${CUDA_HEADERS})
# add_style_check_target(hppl ${HPPL_CU_SOURCES}) # TODO(yuyang18): Format hppl style
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册