257 lines
8.8 KiB
CMake
257 lines
8.8 KiB
CMake
cmake_policy(SET CMP0091 NEW)
|
|
set(CMAKE_POLICY_DEFAULT_CMP0091 NEW)
|
|
set(CMAKE_MSVC_RUNTIME_LIBRARY "MultiThreaded$<$<CONFIG:Debug>:Debug>")
|
|
|
|
project(rife-ncnn-vulkan)
|
|
|
|
cmake_minimum_required(VERSION 3.9)
|
|
|
|
set(CMAKE_BUILD_TYPE Release)
|
|
|
|
option(USE_SYSTEM_NCNN "build with system libncnn" OFF)
|
|
option(USE_SYSTEM_WEBP "build with system libwebp" OFF)
|
|
option(USE_STATIC_MOLTENVK "link moltenvk static library" OFF)
|
|
|
|
find_package(Threads)
|
|
find_package(OpenMP)
|
|
find_package(Vulkan REQUIRED)
|
|
|
|
macro(rife_add_shader SHADER_SRC)
|
|
get_filename_component(SHADER_SRC_NAME_WE ${SHADER_SRC} NAME_WE)
|
|
set(SHADER_COMP_HEADER ${CMAKE_CURRENT_BINARY_DIR}/${SHADER_SRC_NAME_WE}.comp.hex.h)
|
|
|
|
add_custom_command(
|
|
OUTPUT ${SHADER_COMP_HEADER}
|
|
COMMAND ${CMAKE_COMMAND} -DSHADER_SRC=${CMAKE_CURRENT_SOURCE_DIR}/${SHADER_SRC} -DSHADER_COMP_HEADER=${SHADER_COMP_HEADER} -P "${CMAKE_CURRENT_SOURCE_DIR}/generate_shader_comp_header.cmake"
|
|
DEPENDS ${CMAKE_CURRENT_SOURCE_DIR}/${SHADER_SRC}
|
|
COMMENT "Preprocessing shader source ${SHADER_SRC_NAME_WE}.comp"
|
|
VERBATIM
|
|
)
|
|
set_source_files_properties(${SHADER_COMP_HEADER} PROPERTIES GENERATED TRUE)
|
|
|
|
list(APPEND SHADER_SPV_HEX_FILES ${SHADER_COMP_HEADER})
|
|
endmacro()
|
|
|
|
include_directories(${CMAKE_CURRENT_BINARY_DIR})
|
|
|
|
if(OPENMP_FOUND)
|
|
set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} ${OpenMP_C_FLAGS}")
|
|
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${OpenMP_CXX_FLAGS}")
|
|
set(CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} ${OpenMP_EXE_LINKER_FLAGS}")
|
|
endif()
|
|
|
|
# enable global link time optimization
|
|
cmake_policy(SET CMP0069 NEW)
|
|
set(CMAKE_POLICY_DEFAULT_CMP0069 NEW)
|
|
include(CheckIPOSupported)
|
|
check_ipo_supported(RESULT ipo_supported OUTPUT ipo_supported_output)
|
|
if(ipo_supported)
|
|
set(CMAKE_INTERPROCEDURAL_OPTIMIZATION FALSE)
|
|
else()
|
|
message(WARNING "IPO is not supported: ${ipo_supported_output}")
|
|
endif()
|
|
|
|
if(USE_SYSTEM_NCNN)
|
|
set(GLSLANG_TARGET_DIR "GLSLANG-NOTFOUND" CACHE PATH "Absolute path to glslangTargets.cmake directory")
|
|
if(NOT GLSLANG_TARGET_DIR AND NOT DEFINED ENV{GLSLANG_TARGET_DIR})
|
|
message(WARNING "GLSLANG_TARGET_DIR must be defined! USE_SYSTEM_NCNN will be turned off.")
|
|
set(USE_SYSTEM_NCNN OFF)
|
|
else()
|
|
message(STATUS "Using glslang install located at ${GLSLANG_TARGET_DIR}")
|
|
|
|
find_package(Threads)
|
|
|
|
include("${GLSLANG_TARGET_DIR}/OSDependentTargets.cmake")
|
|
include("${GLSLANG_TARGET_DIR}/OGLCompilerTargets.cmake")
|
|
if(EXISTS "${GLSLANG_TARGET_DIR}/HLSLTargets.cmake")
|
|
# hlsl support can be optional
|
|
include("${GLSLANG_TARGET_DIR}/HLSLTargets.cmake")
|
|
endif()
|
|
include("${GLSLANG_TARGET_DIR}/glslangTargets.cmake")
|
|
include("${GLSLANG_TARGET_DIR}/SPIRVTargets.cmake")
|
|
|
|
if (NOT TARGET glslang OR NOT TARGET SPIRV)
|
|
message(WARNING "glslang or SPIRV target not found! USE_SYSTEM_NCNN will be turned off.")
|
|
set(USE_SYSTEM_NCNN OFF)
|
|
endif()
|
|
endif()
|
|
endif()
|
|
|
|
if(USE_SYSTEM_NCNN)
|
|
find_package(ncnn)
|
|
if(NOT TARGET ncnn)
|
|
message(WARNING "ncnn target not found! USE_SYSTEM_NCNN will be turned off.")
|
|
set(USE_SYSTEM_NCNN OFF)
|
|
endif()
|
|
endif()
|
|
|
|
if(NOT USE_SYSTEM_NCNN)
|
|
# build ncnn library
|
|
if(NOT EXISTS "${CMAKE_CURRENT_SOURCE_DIR}/ncnn/CMakeLists.txt")
|
|
message(FATAL_ERROR "The submodules were not downloaded! Please update submodules with \"git submodule update --init --recursive\" and try again.")
|
|
endif()
|
|
|
|
option(NCNN_INSTALL_SDK "" OFF)
|
|
option(NCNN_PIXEL_ROTATE "" OFF)
|
|
option(NCNN_VULKAN "" ON)
|
|
option(NCNN_VULKAN_ONLINE_SPIRV "" ON)
|
|
option(NCNN_BUILD_BENCHMARK "" ON)
|
|
option(NCNN_BUILD_TESTS "" OFF)
|
|
option(NCNN_BUILD_TOOLS "" OFF)
|
|
option(NCNN_BUILD_EXAMPLES "" OFF)
|
|
option(NCNN_DISABLE_RTTI "" ON)
|
|
option(NCNN_DISABLE_EXCEPTION "" ON)
|
|
|
|
option(WITH_LAYER_absval "" OFF)
|
|
option(WITH_LAYER_argmax "" OFF)
|
|
option(WITH_LAYER_batchnorm "" OFF)
|
|
option(WITH_LAYER_bias "" OFF)
|
|
option(WITH_LAYER_bnll "" OFF)
|
|
option(WITH_LAYER_concat "" ON)
|
|
option(WITH_LAYER_convolution "" ON)
|
|
option(WITH_LAYER_crop "" ON)
|
|
option(WITH_LAYER_deconvolution "" ON)
|
|
option(WITH_LAYER_dropout "" OFF)
|
|
option(WITH_LAYER_eltwise "" OFF)
|
|
option(WITH_LAYER_elu "" OFF)
|
|
option(WITH_LAYER_embed "" OFF)
|
|
option(WITH_LAYER_exp "" OFF)
|
|
option(WITH_LAYER_flatten "" ON)
|
|
option(WITH_LAYER_innerproduct "" ON)
|
|
option(WITH_LAYER_input "" ON)
|
|
option(WITH_LAYER_log "" OFF)
|
|
option(WITH_LAYER_lrn "" OFF)
|
|
option(WITH_LAYER_memorydata "" OFF)
|
|
option(WITH_LAYER_mvn "" OFF)
|
|
option(WITH_LAYER_pooling "" ON)
|
|
option(WITH_LAYER_power "" OFF)
|
|
option(WITH_LAYER_prelu "" ON)
|
|
option(WITH_LAYER_proposal "" OFF)
|
|
option(WITH_LAYER_reduction "" OFF)
|
|
option(WITH_LAYER_relu "" ON)
|
|
option(WITH_LAYER_reshape "" OFF)
|
|
option(WITH_LAYER_roipooling "" OFF)
|
|
option(WITH_LAYER_scale "" OFF)
|
|
option(WITH_LAYER_sigmoid "" ON)
|
|
option(WITH_LAYER_slice "" OFF)
|
|
option(WITH_LAYER_softmax "" OFF)
|
|
option(WITH_LAYER_split "" ON)
|
|
option(WITH_LAYER_spp "" OFF)
|
|
option(WITH_LAYER_tanh "" OFF)
|
|
option(WITH_LAYER_threshold "" OFF)
|
|
option(WITH_LAYER_tile "" OFF)
|
|
option(WITH_LAYER_rnn "" OFF)
|
|
option(WITH_LAYER_lstm "" OFF)
|
|
option(WITH_LAYER_binaryop "" ON)
|
|
option(WITH_LAYER_unaryop "" ON)
|
|
option(WITH_LAYER_convolutiondepthwise "" OFF)
|
|
option(WITH_LAYER_padding "" ON)
|
|
option(WITH_LAYER_squeeze "" OFF)
|
|
option(WITH_LAYER_expanddims "" OFF)
|
|
option(WITH_LAYER_normalize "" OFF)
|
|
option(WITH_LAYER_permute "" OFF)
|
|
option(WITH_LAYER_priorbox "" OFF)
|
|
option(WITH_LAYER_detectionoutput "" OFF)
|
|
option(WITH_LAYER_interp "" ON)
|
|
option(WITH_LAYER_deconvolutiondepthwise "" OFF)
|
|
option(WITH_LAYER_shufflechannel "" OFF)
|
|
option(WITH_LAYER_instancenorm "" OFF)
|
|
option(WITH_LAYER_clip "" ON)
|
|
option(WITH_LAYER_reorg "" OFF)
|
|
option(WITH_LAYER_yolodetectionoutput "" OFF)
|
|
option(WITH_LAYER_quantize "" OFF)
|
|
option(WITH_LAYER_dequantize "" OFF)
|
|
option(WITH_LAYER_yolov3detectionoutput "" OFF)
|
|
option(WITH_LAYER_psroipooling "" OFF)
|
|
option(WITH_LAYER_roialign "" OFF)
|
|
option(WITH_LAYER_packing "" ON)
|
|
option(WITH_LAYER_requantize "" OFF)
|
|
option(WITH_LAYER_cast "" ON)
|
|
option(WITH_LAYER_hardsigmoid "" OFF)
|
|
option(WITH_LAYER_selu "" OFF)
|
|
option(WITH_LAYER_hardswish "" OFF)
|
|
option(WITH_LAYER_noop "" OFF)
|
|
option(WITH_LAYER_pixelshuffle "" ON)
|
|
option(WITH_LAYER_deepcopy "" OFF)
|
|
option(WITH_LAYER_mish "" OFF)
|
|
option(WITH_LAYER_statisticspooling "" OFF)
|
|
option(WITH_LAYER_swish "" OFF)
|
|
|
|
add_subdirectory(ncnn)
|
|
endif()
|
|
|
|
if(USE_SYSTEM_WEBP)
|
|
set(CMAKE_MODULE_PATH "${CMAKE_CURRENT_SOURCE_DIR}")
|
|
find_package(WebP)
|
|
if(NOT TARGET webp)
|
|
message(WARNING "webp target not found! USE_SYSTEM_WEBP will be turned off.")
|
|
set(USE_SYSTEM_WEBP OFF)
|
|
endif()
|
|
endif()
|
|
|
|
if(NOT USE_SYSTEM_WEBP)
|
|
# build libwebp library
|
|
if(NOT EXISTS "${CMAKE_CURRENT_SOURCE_DIR}/libwebp/CMakeLists.txt")
|
|
message(FATAL_ERROR "The submodules were not downloaded! Please update submodules with \"git submodule update --init --recursive\" and try again.")
|
|
endif()
|
|
|
|
option(WEBP_ENABLE_SIMD "" ON)
|
|
option(WEBP_BUILD_ANIM_UTILS "" OFF)
|
|
option(WEBP_BUILD_CWEBP "" OFF)
|
|
option(WEBP_BUILD_DWEBP "" OFF)
|
|
option(WEBP_BUILD_GIF2WEBP "" OFF)
|
|
option(WEBP_BUILD_IMG2WEBP "" OFF)
|
|
option(WEBP_BUILD_VWEBP "" OFF)
|
|
option(WEBP_BUILD_WEBPINFO "" OFF)
|
|
option(WEBP_BUILD_WEBPMUX "" OFF)
|
|
option(WEBP_BUILD_EXTRAS "" OFF)
|
|
option(WEBP_BUILD_WEBP_JS "" OFF)
|
|
option(WEBP_NEAR_LOSSLESS "" OFF)
|
|
option(WEBP_ENABLE_SWAP_16BIT_CSP "" OFF)
|
|
|
|
add_subdirectory(libwebp)
|
|
|
|
include_directories(${CMAKE_CURRENT_SOURCE_DIR}/libwebp/src)
|
|
endif()
|
|
|
|
rife_add_shader(rife_preproc.comp)
|
|
rife_add_shader(rife_postproc.comp)
|
|
rife_add_shader(warp.comp)
|
|
rife_add_shader(warp_pack4.comp)
|
|
|
|
add_custom_target(generate-spirv DEPENDS ${SHADER_SPV_HEX_FILES})
|
|
|
|
add_executable(rife-ncnn-vulkan
|
|
main.cpp
|
|
rife.cpp
|
|
warp.cpp
|
|
)
|
|
|
|
add_dependencies(rife-ncnn-vulkan generate-spirv)
|
|
|
|
set(RIFE_LINK_LIBRARIES ncnn webp ${Vulkan_LIBRARY})
|
|
|
|
if(USE_STATIC_MOLTENVK)
|
|
find_library(CoreFoundation NAMES CoreFoundation)
|
|
find_library(Foundation NAMES Foundation)
|
|
find_library(Metal NAMES Metal)
|
|
find_library(QuartzCore NAMES QuartzCore)
|
|
find_library(CoreGraphics NAMES CoreGraphics)
|
|
find_library(Cocoa NAMES Cocoa)
|
|
find_library(IOKit NAMES IOKit)
|
|
find_library(IOSurface NAMES IOSurface)
|
|
|
|
list(APPEND RIFE_LINK_LIBRARIES
|
|
${Metal}
|
|
${QuartzCore}
|
|
${CoreGraphics}
|
|
${Cocoa}
|
|
${IOKit}
|
|
${IOSurface}
|
|
${Foundation}
|
|
${CoreFoundation}
|
|
)
|
|
endif()
|
|
|
|
target_link_libraries(rife-ncnn-vulkan ${RIFE_LINK_LIBRARIES})
|