73 lines
1.4 KiB
CMake
73 lines
1.4 KiB
CMake
set(PIM_ENABLED 1 BOOL PARENT_SCOPE)
|
|
|
|
set(PIM_SRC_ROOT "${CMAKE_CURRENT_SOURCE_DIR}")
|
|
set(PIM_BIN_ROOT "${CMAKE_CURRENT_BINARY_DIR}")
|
|
|
|
set(PIM_LIBRARY_PATH ${CMAKE_LIBRARY_OUTPUT_DIRECTORY})
|
|
set(PIM_RUNTIME_PATH ${CMAKE_RUNTIME_OUTPUT_DIRECTORY})
|
|
set(PIM_INCLUDE_PATH ${CMAKE_INCLUDE_OUTPUT_DIRECTORY})
|
|
|
|
set(PIM_ONNX_MLIR_SRC_ROOT ${ONNX_MLIR_SRC_ROOT})
|
|
set(PIM_ONNX_MLIR_BIN_ROOT ${ONNX_MLIR_BIN_ROOT})
|
|
|
|
set(PIM_PUBLIC_INCLUDE_DIRS
|
|
${ONNX_MLIR_SRC_ROOT}/include
|
|
${ONNX_MLIR_SRC_ROOT}
|
|
${PIM_ONNX_MLIR_SRC_ROOT}
|
|
${PIM_SRC_ROOT}
|
|
${PIM_BIN_ROOT}
|
|
${PIM_INCLUDE_PATH}
|
|
)
|
|
|
|
set(PIM_COMPILER_INCLUDE_DIRS
|
|
${PIM_SRC_ROOT}
|
|
${PIM_BIN_ROOT}
|
|
${PIM_ONNX_MLIR_SRC_ROOT}
|
|
${PIM_ONNX_MLIR_BIN_ROOT}
|
|
)
|
|
|
|
set(PIM_ACCEL_INCLUDE_DIRS
|
|
${PIM_ONNX_MLIR_SRC_ROOT}
|
|
${PIM_ONNX_MLIR_BIN_ROOT}
|
|
)
|
|
|
|
set(PIM_GENERATED_INCLUDE_DIRS
|
|
${PIM_INCLUDE_PATH}
|
|
)
|
|
|
|
function(add_pim_library name)
|
|
add_onnx_mlir_library(${name} STATIC ${ARGN})
|
|
endfunction()
|
|
|
|
add_subdirectory(Dialect)
|
|
add_subdirectory(Common)
|
|
add_subdirectory(Pass)
|
|
add_subdirectory(Compiler)
|
|
add_subdirectory(Conversion)
|
|
|
|
add_pim_library(OMPIMAccel
|
|
PimAccelerator.cpp
|
|
|
|
EXCLUDE_FROM_OM_LIBS
|
|
|
|
INCLUDE_DIRS PUBLIC
|
|
${PIM_PUBLIC_INCLUDE_DIRS}
|
|
|
|
LINK_LIBS PUBLIC
|
|
MLIRSCFDialect
|
|
MLIRSCFTransforms
|
|
onnx
|
|
OMAccelerator
|
|
OMPimCompilerUtils
|
|
OMPimPasses
|
|
OMONNXOps
|
|
SpatialOps
|
|
PimOps
|
|
OMONNXToSpatial
|
|
OMSpatialToGraphviz
|
|
OMSpatialToPim
|
|
OMPimCommon
|
|
OMPimBufferization
|
|
MLIRTensorInferTypeOpInterfaceImpl
|
|
)
|