mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
b9adbb5002
Summary: Ignore mixed upper-case/lower-case style for now Fix space between function and its arguments violation Pull Request resolved: https://github.com/pytorch/pytorch/pull/35574 Test Plan: CI Differential Revision: D20712969 Pulled By: malfet fbshipit-source-id: 0012d430aed916b4518599a0b535e82d15721f78
125 lines
3.0 KiB
CMake
125 lines
3.0 KiB
CMake
cmake_minimum_required(VERSION 3.4.1)
|
|
project(pytorch_jni CXX)
|
|
set(CMAKE_CXX_STANDARD 14)
|
|
set(CMAKE_VERBOSE_MAKEFILE ON)
|
|
|
|
set(TRACE_ENABLED OFF)
|
|
if(DEFINED ENV{TRACE_ENABLED})
|
|
if($ENV{TRACE_ENABLED} STREQUAL "1")
|
|
message(STATUS "TRACE_ENABLED ON")
|
|
set(TRACE_ENABLED ON)
|
|
endif()
|
|
endif()
|
|
if(NOT TRACE_ENABLED)
|
|
message(STATUS "TRACE_ENABLED OFF")
|
|
endif()
|
|
|
|
set(pytorch_android_DIR ${CMAKE_CURRENT_LIST_DIR}/src/main/cpp)
|
|
|
|
if(ANDROID_ABI)
|
|
set(libtorch_include_DIR ${pytorch_android_DIR}/libtorch_include/${ANDROID_ABI})
|
|
set(BUILD_SUBDIR ${ANDROID_ABI})
|
|
elseif(BUILD_LIBTORCH_WITH_JNI)
|
|
# Don't need LIBTORCH_HOME if we're building from within PyTorch.
|
|
else()
|
|
# Building against a pre-built libtorch.
|
|
if(NOT LIBTORCH_HOME)
|
|
message(FATAL_ERROR
|
|
"pytorch_android requires LIBTORCH_HOME to be defined for non-Android builds.")
|
|
endif()
|
|
set(libtorch_include_DIR ${LIBTORCH_HOME}/include)
|
|
link_directories(${LIBTORCH_HOME}/lib)
|
|
set(BUILD_SUBDIR host)
|
|
endif()
|
|
|
|
message(STATUS "libtorch dir:${libtorch_DIR}")
|
|
|
|
configure_file(
|
|
${pytorch_android_DIR}/cmake_macros.h.in
|
|
${pytorch_android_DIR}/cmake_macros.h)
|
|
|
|
file(GLOB pytorch_android_SOURCES
|
|
${pytorch_android_DIR}/pytorch_jni_jit.cpp
|
|
${pytorch_android_DIR}/pytorch_jni_common.cpp
|
|
${pytorch_android_DIR}/pytorch_jni_common.h
|
|
)
|
|
|
|
add_library(pytorch_jni SHARED
|
|
${pytorch_android_SOURCES}
|
|
)
|
|
|
|
if(APPLE)
|
|
# Need to add rpath so dlopen can find dependencies.
|
|
add_custom_command(TARGET pytorch_jni
|
|
POST_BUILD COMMAND
|
|
${CMAKE_INSTALL_NAME_TOOL} -add_rpath "@loader_path"
|
|
$<TARGET_FILE:pytorch_jni>)
|
|
endif()
|
|
|
|
target_compile_options(pytorch_jni PRIVATE
|
|
-fexceptions
|
|
)
|
|
|
|
target_include_directories(pytorch_jni PUBLIC
|
|
${libtorch_include_DIR}
|
|
)
|
|
|
|
set(fbjni_DIR ${CMAKE_CURRENT_LIST_DIR}/../libs/fbjni/)
|
|
set(fbjni_BUILD_DIR ${CMAKE_BINARY_DIR}/fbjni/${BUILD_SUBDIR})
|
|
|
|
add_subdirectory(${fbjni_DIR} ${fbjni_BUILD_DIR})
|
|
|
|
if(ANDROID_ABI)
|
|
|
|
function(import_static_lib name)
|
|
add_library(${name} STATIC IMPORTED)
|
|
set_property(
|
|
TARGET ${name}
|
|
PROPERTY IMPORTED_LOCATION
|
|
${CMAKE_CURRENT_LIST_DIR}/src/main/jniLibs/${ANDROID_ABI}/${name}.a)
|
|
endfunction(import_static_lib)
|
|
|
|
import_static_lib(libtorch)
|
|
import_static_lib(libtorch_cpu)
|
|
import_static_lib(libc10)
|
|
import_static_lib(libnnpack)
|
|
import_static_lib(libXNNPACK)
|
|
import_static_lib(libpytorch_qnnpack)
|
|
import_static_lib(libeigen_blas)
|
|
import_static_lib(libcpuinfo)
|
|
import_static_lib(libclog)
|
|
|
|
# Link most things statically on Android.
|
|
target_link_libraries(pytorch_jni
|
|
fbjni
|
|
-Wl,--gc-sections
|
|
-Wl,--whole-archive
|
|
libtorch
|
|
libtorch_cpu
|
|
-Wl,--no-whole-archive
|
|
libc10
|
|
libnnpack
|
|
libXNNPACK
|
|
libpytorch_qnnpack
|
|
libeigen_blas
|
|
libcpuinfo
|
|
libclog
|
|
)
|
|
|
|
else()
|
|
|
|
# Prefer dynamic linking on the host
|
|
target_link_libraries(pytorch_jni
|
|
fbjni
|
|
torch
|
|
torch_cpu
|
|
c10
|
|
nnpack
|
|
XNNPACK
|
|
pytorch_qnnpack
|
|
cpuinfo
|
|
clog
|
|
)
|
|
|
|
endif()
|