1if(NOT __AOTRITON_INCLUDED) 2 set(__AOTRITON_INCLUDED TRUE) 3 4 set(__AOTRITON_SOURCE_DIR "${CMAKE_CURRENT_BINARY_DIR}/aotriton/src") 5 set(__AOTRITON_BUILD_DIR "${CMAKE_CURRENT_BINARY_DIR}/aotriton/build") 6 set(__AOTRITON_INSTALL_DIR "${PROJECT_SOURCE_DIR}/torch") 7 add_library(__caffe2_aotriton INTERFACE) 8 # Note it is INSTALL"ED" 9 if(DEFINED ENV{AOTRITON_INSTALLED_PREFIX}) 10 set(__AOTRITON_INSTALL_DIR "$ENV{AOTRITON_INSTALLED_PREFIX}") 11 message(STATUS "Using Preinstalled AOTriton at ${__AOTRITON_INSTALL_DIR}") 12 else() 13 file(STRINGS "${CMAKE_CURRENT_SOURCE_DIR}/.ci/docker/aotriton_version.txt" __AOTRITON_CI_INFO) 14 list(GET __AOTRITON_CI_INFO 3 __AOTRITON_CI_COMMIT) 15 ExternalProject_Add(aotriton_external 16 GIT_REPOSITORY https://github.com/ROCm/aotriton.git 17 GIT_TAG ${__AOTRITON_CI_COMMIT} 18 SOURCE_DIR ${__AOTRITON_SOURCE_DIR} 19 BINARY_DIR ${__AOTRITON_BUILD_DIR} 20 PREFIX ${__AOTRITON_INSTALL_DIR} 21 CMAKE_ARGS -DCMAKE_INSTALL_PREFIX:PATH=${__AOTRITON_INSTALL_DIR} 22 -DAOTRITON_COMPRESS_KERNEL=OFF 23 -DCMAKE_BUILD_TYPE=${CMAKE_BUILD_TYPE} 24 -DAOTRITON_NO_PYTHON=ON 25 -DAOTRITON_NO_SHARED=OFF 26 # CONFIGURE_COMMAND "" 27 BUILD_COMMAND "" # No build, install command will repeat the build process due to problems in the build system. 28 BUILD_BYPRODUCTS "${__AOTRITON_INSTALL_DIR}/lib/libaotriton_v2.so" 29 USES_TERMINAL_DOWNLOAD TRUE 30 USES_TERMINAL_CONFIGURE TRUE 31 USES_TERMINAL_BUILD TRUE 32 USES_TERMINAL_INSTALL TRUE 33 # INSTALL_COMMAND ${MAKE_COMMAND} install 34 ) 35 add_dependencies(__caffe2_aotriton aotriton_external) 36 message(STATUS "Using AOTriton compiled from source directory ${__AOTRITON_SOURCE_DIR}") 37 endif() 38 target_link_libraries(__caffe2_aotriton INTERFACE ${__AOTRITON_INSTALL_DIR}/lib/libaotriton_v2.so) 39 target_include_directories(__caffe2_aotriton INTERFACE ${__AOTRITON_INSTALL_DIR}/include) 40 set(AOTRITON_FOUND TRUE) 41endif() # __AOTRITON_INCLUDED 42