chore: import upstream snapshot with attribution
This commit is contained in:
Executable
+38
@@ -0,0 +1,38 @@
|
||||
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})
|
||||
Reference in New Issue
Block a user