39 lines
1.1 KiB
CMake
39 lines
1.1 KiB
CMake
|
if(LINUX)
|
||
|
|
link_directories(/usr/local/lib)
|
||
|
|
link_directories(/usr/lib)
|
||
|
|
endif()
|
||
|
|
|
||
|
|
include_directories(../include)
|
||
|
|
#used for determining if we should include jni components or not
|
||
|
|
find_package(Java COMPONENTS Development)
|
||
|
|
if(Java_FOUND)
|
||
|
|
if(EXISTS "$ENV{JAVA_HOME}")
|
||
|
|
message("INCLUDING JAVA $ENV{JAVA_HOME}/include")
|
||
|
|
include_directories("$ENV{JAVA_HOME}/include" "$ENV{JAVA_HOME}/include/linux")
|
||
|
|
add_definitions(-D__JNI__=true)
|
||
|
|
|
||
|
|
else()
|
||
|
|
message(FATAL_ERROR "Could not find JAVA_HOME environment variable")
|
||
|
|
endif(EXISTS "$ENV{JAVA_HOME}")
|
||
|
|
endif()
|
||
|
|
|
||
|
|
FIND_PACKAGE(BLAS)
|
||
|
|
FIND_PACKAGE(LAPACK)
|
||
|
|
|
||
|
|
if(NOT BLAS_FOUND)
|
||
|
|
message("BLAS not found using openblas fallback")
|
||
|
|
set(BLAS_LIBRARIES /usr/lib/libopenblas.a)
|
||
|
|
FIND_PACKAGE(BLAS REQUIRED)
|
||
|
|
endif()
|
||
|
|
if(NOT LAPACK_FOUND)
|
||
|
|
message("LAPACK not found using openblas fallback")
|
||
|
|
set(LAPACK_LIBRARIES /usr/lib/libopenblas.a)
|
||
|
|
FIND_PACKAGE(LAPACK REQUIRED)
|
||
|
|
endif()
|
||
|
|
|
||
|
|
|
||
|
|
SET( CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}" )
|
||
|
|
|
||
|
|
add_library(libnd4j SHARED blasimpl.cpp NativeOps.cpp)
|
||
|
|
target_link_libraries(libnd4j ${BLAS_LIBRARIES} ${LAPACK_LIBRARIES})
|