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