e25f20f7a3
commit 3f4938648950a7f3bf9a19c320ca9fae7c52de20 Author: sophgo-forum-service <forum_service@sophgo.com> Date: Mon May 13 13:44:23 2024 +0800 [feat] cviruntime opensource for cv18xx soc. - a4b6a3, add cumsum and gatherelements_pt.
22 lines
574 B
CMake
22 lines
574 B
CMake
cmake_minimum_required(VERSION 2.8.0)
|
|
project(custom_cpu_function CXX)
|
|
|
|
set(CMAKE_CXX_STANDARD 17)
|
|
set(CMAKE_CXX_STANDARD_REQUIRED ON)
|
|
set(CMAKE_CXX_EXTENSIONS ON)
|
|
|
|
if(NOT DEFINED MLIR_INCLUDE)
|
|
message(FATAL_ERROR "Please set MLIR_INCLUDE to point to the include path of mlir")
|
|
endif()
|
|
|
|
include_directories(${PROJECT_SOURCE_DIR})
|
|
include_directories(${MLIR_INCLUDE})
|
|
|
|
add_library(CustomOpPlugin SHARED
|
|
LeakyReluOp.cpp
|
|
ROIAlignOp.cpp
|
|
SoftmaxOp.cpp
|
|
UnPoolingOp.cpp)
|
|
|
|
install(TARGETS CustomOpPlugin DESTINATION lib/custom_op/)
|