File tree 4 files changed +4
-55
lines changed
4 files changed +4
-55
lines changed Original file line number Diff line number Diff line change @@ -150,7 +150,6 @@ if(WITH_TESTING)
150
150
include_directories (${GTEST_INCLUDE_DIRS} )
151
151
endif ()
152
152
153
- include_directories ("${ATLAS_INC_DIR} /atlas" )
154
153
include_directories ("${CBLAS_INC_DIR} " )
155
154
include_directories ("${PROJ_ROOT} " )
156
155
include_directories ("${PROJ_ROOT} /paddle/cuda/include" )
Load Diff This file was deleted.
Original file line number Diff line number Diff line change @@ -58,15 +58,16 @@ set(ATLAS_LIB_SEARCH_PATHS
58
58
)
59
59
find_path (ATLAS_INC_DIR NAMES cblas.h
60
60
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} )
62
63
find_library (ATLAS_CBLAS_LIB NAMES cblas libcblas.so.3
63
64
PATHS ${ATLAS_LIB_SEARCH_PATHS} )
64
65
find_library (ATLAS_LIB NAMES lapack_atlas liblapack_atlas.so.3
65
66
PATHS ${ATLAS_LIB_SEARCH_PATHS} )
66
67
67
68
if (ATLAS_INC_DIR AND ATLAS_CBLAS_LIB AND ATLAS_LIB)
68
69
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 } )
70
71
set (CBLAS_LIBS ${ATLAS_LIB} ${ATLAS_CBLAS_LIB} )
71
72
add_definitions (-DPADDLE_USE_ATLAS)
72
73
message (STATUS "Found Atlas (include: ${CBLAS_INC_DIR} , library: ${CBLAS_LIBS} )" )
Original file line number Diff line number Diff line change @@ -17,6 +17,7 @@ limitations under the License. */
17
17
18
18
#ifdef PADDLE_USE_MKL
19
19
#include < mkl.h>
20
+ #include < mkl_lapacke.h>
20
21
#else
21
22
extern " C" {
22
23
#include < cblas.h>
You can’t perform that action at this time.
0 commit comments