|
19 | 19 | # THE SOFTWARE.
|
20 | 20 |
|
21 | 21 | cmake_minimum_required(VERSION 3.10)
|
| 22 | + |
22 | 23 | set(VERSION "1.0.1")
|
23 | 24 | project(vx_amd_custom VERSION ${VERSION} LANGUAGES CXX)
|
24 | 25 |
|
25 | 26 | set(CMAKE_CXX_STANDARD 14)
|
26 | 27 |
|
27 |
| -list (APPEND CMAKE_PREFIX_PATH ${ROCM_PATH} ${ROCM_PATH}/hip) |
| 28 | +# ROCM Path |
| 29 | +if(DEFINED ENV{ROCM_PATH}) |
| 30 | + set(ROCM_PATH $ENV{ROCM_PATH} CACHE PATH "Default ROCm installation path") |
| 31 | +elseif(ROCM_PATH) |
| 32 | + message("-- ${PROJECT_NAME}: INFO -- ROCM_PATH Set -- ${ROCM_PATH}") |
| 33 | +else() |
| 34 | + set(ROCM_PATH /opt/rocm CACHE PATH "Default ROCm installation path") |
| 35 | +endif() |
28 | 36 |
|
29 |
| -# OpenMP |
30 |
| -find_package(OpenMP QUIET) |
| 37 | +list(APPEND CMAKE_PREFIX_PATH ${ROCM_PATH} ${ROCM_PATH}/hip) |
31 | 38 |
|
32 |
| -if(GPU_SUPPORT AND "${BACKEND}" STREQUAL "HIP" AND OpenMP_FOUND) |
| 39 | +# Find Packages |
| 40 | +# HIP |
| 41 | +if(GPU_SUPPORT AND "${BACKEND}" STREQUAL "HIP") |
33 | 42 | if(NOT DEFINED HIP_PATH)
|
34 | 43 | if(NOT DEFINED ENV{HIP_PATH})
|
35 |
| - set(HIP_PATH ${ROCM_PATH}) |
| 44 | + set(HIP_PATH ${ROCM_PATH} CACHE PATH "Path to which HIP has been installed") |
36 | 45 | else()
|
37 |
| - set(HIP_PATH $ENV{HIP_PATH}) |
| 46 | + set(HIP_PATH $ENV{HIP_PATH} CACHE PATH "Path to which HIP has been installed") |
38 | 47 | endif()
|
39 | 48 | endif()
|
40 | 49 | list(APPEND CMAKE_PREFIX_PATH ${ROCM_PATH} ${ROCM_PATH}/hip)
|
41 | 50 | find_package(HIP QUIET)
|
| 51 | + if(NOT HIP_FOUND) |
| 52 | + message("-- ${Red}${PROJECT_NAME} WARNING: HIP Not Found${ColourReset}") |
| 53 | + endif() |
| 54 | +endif() |
| 55 | +# OpenMP |
| 56 | +find_package(OpenMP QUIET) |
| 57 | +if(OpenMP_FOUND) |
42 | 58 | set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} ${OpenMP_C_FLAGS}")
|
43 | 59 | set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${OpenMP_CXX_FLAGS}")
|
| 60 | + include_directories(${OpenMP_INCLUDE_DIRS}) |
44 | 61 | set(LINK_LIBRARY_LIST ${LINK_LIBRARY_LIST} OpenMP::OpenMP_CXX)
|
| 62 | + message("-- ${White}${PROJECT_NAME} -- OpenMP Found${ColourReset}") |
| 63 | +else() |
| 64 | + message("-- ${Yellow}WARNING: ${PROJECT_NAME} -- OpenMP Not FOUND${ColourReset}") |
| 65 | +endif() |
45 | 66 |
|
46 |
| - if("${CMAKE_CXX_COMPILER_ID}" STREQUAL "MSVC") |
47 |
| - set(CMAKE_CXX_FLAGS_RELEASE "${CMAKE_CXX_FLAGS_RELEASE} /MD") |
48 |
| - set(CMAKE_CXX_FLAGS_DEBUG "${CMAKE_CXX_FLAGS_DEBUG} /MDd") |
49 |
| - else() |
50 |
| - set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -fPIC -msse4.2") |
51 |
| - endif() |
| 67 | +list(APPEND SOURCES source/custom_kernels.cpp source/custom_layer.cpp) |
| 68 | +include_directories(../../amd_openvx/openvx/include ${ROCM_PATH}/include include ./custom_lib) |
| 69 | +set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -msse4.2") |
52 | 70 |
|
| 71 | +if(HIP_FOUND AND OpenMP_FOUND) |
| 72 | + message("-- ${Green}${PROJECT_NAME} -- Building HIP backend${ColourReset}") |
| 73 | + set(ENABLE_HIP 1) |
| 74 | + link_directories(${HIP_PATH}/lib) |
53 | 75 | list(APPEND PACKAGE_DEPENDS PACKAGE HIP)
|
54 |
| - |
55 |
| - include_directories(../../amd_openvx/openvx/include ${ROCM_PATH}/include include ./custom_lib) |
56 |
| - |
57 |
| - list(APPEND SOURCES |
58 |
| - source/custom_kernels.cpp |
59 |
| - source/custom_layer.cpp |
60 |
| - ) |
61 |
| - |
62 |
| - if (HIP_FOUND) |
63 |
| - message("-- ${Green}${PROJECT_NAME} -- Building with HIP backend${ColourReset}") |
64 |
| - set(ENABLE_HIP 1) |
65 |
| - add_definitions(-DENABLE_HIP=${ENABLE_HIP} -D__HIP_PLATFORM_AMD__) |
66 |
| - link_directories(${HIP_PATH}/lib) |
67 |
| - add_subdirectory(custom_lib) |
68 |
| - add_library(${PROJECT_NAME} SHARED ${SOURCES} $<TARGET_OBJECTS:custom_lib>) |
69 |
| - target_link_libraries(${PROJECT_NAME} openvx hip::host ${LINK_LIBRARY_LIST}) |
70 |
| - set_target_properties(${PROJECT_NAME} PROPERTIES VERSION ${PROJECT_VERSION} SOVERSION ${PROJECT_VERSION_MAJOR}) |
71 |
| - # install MIVisionX libs -- {ROCM_PATH}/lib -- TBD: Remove duplicate install |
72 |
| - install(TARGETS ${PROJECT_NAME} LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR} COMPONENT runtime NAMELINK_SKIP) |
73 |
| - install(TARGETS ${PROJECT_NAME} LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR} COMPONENT dev NAMELINK_ONLY) |
74 |
| - install(TARGETS ${PROJECT_NAME} LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR} COMPONENT asan) |
75 |
| - # install MIVisionX include files -- {ROCM_PATH}/include/mivisionx/ |
76 |
| - install(FILES include/vx_amd_custom.h DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}/mivisionx COMPONENT dev) |
77 |
| - else() |
78 |
| - message("-- ${Red}WARNING: HIP Not Found -- ${PROJECT_NAME} module excluded${ColourReset}") |
79 |
| - endif() |
80 |
| -else() |
81 |
| - set(ENABLE_HIP 0) |
82 |
| - add_definitions(-DENABLE_HIP=${ENABLE_HIP}) |
83 |
| - message("-- ${Red}WARNING: GPU Support Turned OFF -- ${PROJECT_NAME} built for CPU only${ColourReset}") |
84 | 76 | add_subdirectory(custom_lib)
|
85 | 77 | add_library(${PROJECT_NAME} SHARED ${SOURCES} $<TARGET_OBJECTS:custom_lib>)
|
86 |
| - target_link_libraries(${PROJECT_NAME} openvx hip::host) |
| 78 | + target_compile_definitions(${PROJECT_NAME} PRIVATE __HIP_PLATFORM_AMD__) |
| 79 | + target_link_libraries(${PROJECT_NAME} openvx hip::host ${LINK_LIBRARY_LIST}) |
| 80 | + target_compile_definitions(${PROJECT_NAME} PUBLIC ENABLE_HIP=${ENABLE_HIP}) |
87 | 81 | set_target_properties(${PROJECT_NAME} PROPERTIES VERSION ${PROJECT_VERSION} SOVERSION ${PROJECT_VERSION_MAJOR})
|
88 |
| - # install MIVisionX libs -- {ROCM_PATH}/lib |
| 82 | + # install MIVisionX libs -- {ROCM_PATH}/lib -- TBD: Remove duplicate install |
89 | 83 | install(TARGETS ${PROJECT_NAME} LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR} COMPONENT runtime NAMELINK_SKIP)
|
90 | 84 | install(TARGETS ${PROJECT_NAME} LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR} COMPONENT dev NAMELINK_ONLY)
|
91 | 85 | install(TARGETS ${PROJECT_NAME} LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR} COMPONENT asan)
|
92 |
| - # install include files -- {ROCM_PATH}/include/mivisionx/ |
| 86 | + # install MIVisionX include files -- {ROCM_PATH}/include/mivisionx/ |
93 | 87 | install(FILES include/vx_amd_custom.h DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}/mivisionx COMPONENT dev)
|
| 88 | +else() |
| 89 | + message("-- ${Red}WARNING:${PROJECT_NAME} -- CPU backend not supported${ColourReset}") |
94 | 90 | endif()
|
0 commit comments