Skip to content

Commit 0d88a88

Browse files
committed
Fix atlas incude dir
1 parent 6e0ec6b commit 0d88a88

File tree

4 files changed

+4
-55
lines changed

4 files changed

+4
-55
lines changed

CMakeLists.txt

-1
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,6 @@ if(WITH_TESTING)
150150
include_directories(${GTEST_INCLUDE_DIRS})
151151
endif()
152152

153-
include_directories("${ATLAS_INC_DIR}/atlas")
154153
include_directories("${CBLAS_INC_DIR}")
155154
include_directories("${PROJ_ROOT}")
156155
include_directories("${PROJ_ROOT}/paddle/cuda/include")

cmake/FindAtlas.cmake

-52
This file was deleted.

cmake/cblas.cmake

+3-2
Original file line numberDiff line numberDiff line change
@@ -58,15 +58,16 @@ set(ATLAS_LIB_SEARCH_PATHS
5858
)
5959
find_path(ATLAS_INC_DIR NAMES cblas.h
6060
PATHS ${ATLAS_INCLUDE_SEARCH_PATHS})
61-
find_path(Atlas_CLAPACK_INCLUDE_DIR NAMES clapack.h PATHS ${Atlas_INCLUDE_SEARCH_PATHS})
61+
find_path(ATLAS_CLAPACK_INC_DIR NAMES clapack.h
62+
PATHS ${ATLAS_INCLUDE_SEARCH_PATHS})
6263
find_library(ATLAS_CBLAS_LIB NAMES cblas libcblas.so.3
6364
PATHS ${ATLAS_LIB_SEARCH_PATHS})
6465
find_library(ATLAS_LIB NAMES lapack_atlas liblapack_atlas.so.3
6566
PATHS ${ATLAS_LIB_SEARCH_PATHS})
6667

6768
if(ATLAS_INC_DIR AND ATLAS_CBLAS_LIB AND ATLAS_LIB)
6869
set(CBLAS_PROVIDER ATLAS)
69-
set(CBLAS_INC_DIR ${ATLAS_INC_DIR} ${Atlas_CLAPACK_INCLUDE_DIR})
70+
set(CBLAS_INC_DIR ${ATLAS_INC_DIR} ${ATLAS_CLAPACK_INC_DIR})
7071
set(CBLAS_LIBS ${ATLAS_LIB} ${ATLAS_CBLAS_LIB})
7172
add_definitions(-DPADDLE_USE_ATLAS)
7273
message(STATUS "Found Atlas (include: ${CBLAS_INC_DIR}, library: ${CBLAS_LIBS})")

paddle/math/MathFunctions.h

+1
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@ limitations under the License. */
1717

1818
#ifdef PADDLE_USE_MKL
1919
#include <mkl.h>
20+
#include <mkl_lapacke.h>
2021
#else
2122
extern "C" {
2223
#include <cblas.h>

0 commit comments

Comments
 (0)