cavis/libnd4j/profile/CMakeLists.txt

39 lines
1.1 KiB
CMake
Executable File

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})