agunashe
a4ff63a29b
Change-Id: I2cb7bbd9a6d9da28116ba9dd9cec4e60525444e2
[ROCm/hip commit: d529637fbc]
195 linhas
7.5 KiB
CMake
195 linhas
7.5 KiB
CMake
# Copyright (c) 2016 - 2021 Advanced Micro Devices, Inc. All Rights Reserved.
|
|
#
|
|
# Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
# of this software and associated documentation files (the "Software"), to deal
|
|
# in the Software without restriction, including without limitation the rights
|
|
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
# copies of the Software, and to permit persons to whom the Software is
|
|
# furnished to do so, subject to the following conditions:
|
|
#
|
|
# The above copyright notice and this permission notice shall be included in
|
|
# all copies or substantial portions of the Software.
|
|
#
|
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
# THE SOFTWARE.
|
|
|
|
###############################################################################
|
|
# Runs commands using HIPCC
|
|
###############################################################################
|
|
|
|
###############################################################################
|
|
# This file runs the hipcc commands to produce the desired output file
|
|
# along with the dependency file needed by CMake to compute dependencies.
|
|
#
|
|
# Input variables:
|
|
#
|
|
# verbose:BOOL=<> OFF: Be as quiet as possible (default)
|
|
# ON : Describe each step
|
|
# build_configuration:STRING=<> Build configuration. Defaults to Debug.
|
|
# generated_file:STRING=<> File to generate. Mandatory argument.
|
|
|
|
if(NOT build_configuration)
|
|
set(build_configuration Debug)
|
|
endif()
|
|
if(NOT generated_file)
|
|
message(FATAL_ERROR "You must specify generated_file on the command line")
|
|
endif()
|
|
|
|
# Set these up as variables to make reading the generated file easier
|
|
set(HIP_HIPCC_EXECUTABLE "@HIP_HIPCC_EXECUTABLE@") # path
|
|
set(HIP_HIPCONFIG_EXECUTABLE "@HIP_HIPCONFIG_EXECUTABLE@") #path
|
|
set(HIP_HOST_COMPILER "@HIP_HOST_COMPILER@") # path
|
|
set(CMAKE_COMMAND "@CMAKE_COMMAND@") # path
|
|
set(HIP_run_make2cmake "@HIP_run_make2cmake@") # path
|
|
set(HIP_CLANG_PATH "@HIP_CLANG_PATH@") #path
|
|
set(HIP_CLANG_PARALLEL_BUILD_COMPILE_OPTIONS "@HIP_CLANG_PARALLEL_BUILD_COMPILE_OPTIONS@")
|
|
|
|
@HIP_HOST_FLAGS@
|
|
@_HIP_HIPCC_FLAGS@
|
|
@_HIP_CLANG_FLAGS@
|
|
@_HIP_NVCC_FLAGS@
|
|
#Needed to bring the HIP_HIPCC_INCLUDE_ARGS variable in scope
|
|
set(HIP_HIPCC_INCLUDE_ARGS @HIP_HIPCC_INCLUDE_ARGS@) # list
|
|
|
|
set(cmake_dependency_file "@cmake_dependency_file@") # path
|
|
set(source_file "@source_file@") # path
|
|
set(host_flag "@host_flag@") # bool
|
|
|
|
# Determine compiler and compiler flags
|
|
execute_process(COMMAND ${HIP_HIPCONFIG_EXECUTABLE} --platform OUTPUT_VARIABLE HIP_PLATFORM OUTPUT_STRIP_TRAILING_WHITESPACE)
|
|
execute_process(COMMAND ${HIP_HIPCONFIG_EXECUTABLE} --compiler OUTPUT_VARIABLE HIP_COMPILER OUTPUT_STRIP_TRAILING_WHITESPACE)
|
|
execute_process(COMMAND ${HIP_HIPCONFIG_EXECUTABLE} --runtime OUTPUT_VARIABLE HIP_RUNTIME OUTPUT_STRIP_TRAILING_WHITESPACE)
|
|
if(NOT host_flag)
|
|
set(__CC ${HIP_HIPCC_EXECUTABLE})
|
|
if("${HIP_PLATFORM}" STREQUAL "amd")
|
|
if("${HIP_COMPILER}" STREQUAL "clang")
|
|
if(NOT "x${HIP_CLANG_PATH}" STREQUAL "x")
|
|
set(ENV{HIP_CLANG_PATH} ${HIP_CLANG_PATH})
|
|
endif()
|
|
set(__CC_FLAGS ${HIP_CLANG_PARALLEL_BUILD_COMPILE_OPTIONS} ${HIP_HIPCC_FLAGS} ${HIP_CLANG_FLAGS} ${HIP_HIPCC_FLAGS_${build_configuration}} ${HIP_CLANG_FLAGS_${build_configuration}})
|
|
endif()
|
|
else()
|
|
set(__CC_FLAGS ${HIP_HIPCC_FLAGS} ${HIP_NVCC_FLAGS} ${HIP_HIPCC_FLAGS_${build_configuration}} ${HIP_NVCC_FLAGS_${build_configuration}})
|
|
endif()
|
|
else()
|
|
set(__CC ${HIP_HOST_COMPILER})
|
|
set(__CC_FLAGS ${CMAKE_HOST_FLAGS} ${CMAKE_HOST_FLAGS_${build_configuration}})
|
|
endif()
|
|
set(__CC_INCLUDES ${HIP_HIPCC_INCLUDE_ARGS})
|
|
|
|
# hip_execute_process - Executes a command with optional command echo and status message.
|
|
# status - Status message to print if verbose is true
|
|
# command - COMMAND argument from the usual execute_process argument structure
|
|
# ARGN - Remaining arguments are the command with arguments
|
|
# HIP_result - Return value from running the command
|
|
macro(hip_execute_process status command)
|
|
set(_command ${command})
|
|
if(NOT "x${_command}" STREQUAL "xCOMMAND")
|
|
message(FATAL_ERROR "Malformed call to hip_execute_process. Missing COMMAND as second argument. (command = ${command})")
|
|
endif()
|
|
if(verbose)
|
|
execute_process(COMMAND "${CMAKE_COMMAND}" -E echo -- ${status})
|
|
# Build command string to print
|
|
set(hip_execute_process_string)
|
|
foreach(arg ${ARGN})
|
|
# Escape quotes if any
|
|
string(REPLACE "\"" "\\\"" arg ${arg})
|
|
# Surround args with spaces with quotes
|
|
if(arg MATCHES " ")
|
|
list(APPEND hip_execute_process_string "\"${arg}\"")
|
|
else()
|
|
list(APPEND hip_execute_process_string ${arg})
|
|
endif()
|
|
endforeach()
|
|
# Echo the command
|
|
execute_process(COMMAND ${CMAKE_COMMAND} -E echo ${hip_execute_process_string})
|
|
endif()
|
|
# Run the command
|
|
execute_process(COMMAND ${ARGN} RESULT_VARIABLE HIP_result)
|
|
endmacro()
|
|
|
|
# Delete the target file
|
|
hip_execute_process(
|
|
"Removing ${generated_file}"
|
|
COMMAND "${CMAKE_COMMAND}" -E remove "${generated_file}"
|
|
)
|
|
|
|
# Generate the dependency file
|
|
hip_execute_process(
|
|
"Generating dependency file: ${cmake_dependency_file}.pre"
|
|
COMMAND "${__CC}"
|
|
-M
|
|
"${source_file}"
|
|
-o "${cmake_dependency_file}.pre"
|
|
${__CC_FLAGS}
|
|
${__CC_INCLUDES}
|
|
)
|
|
|
|
if(HIP_result)
|
|
message(FATAL_ERROR "Error generating ${generated_file}")
|
|
endif()
|
|
|
|
# Generate the cmake readable dependency file to a temp file
|
|
hip_execute_process(
|
|
"Generating temporary cmake readable file: ${cmake_dependency_file}.tmp"
|
|
COMMAND "${CMAKE_COMMAND}"
|
|
-D "input_file:FILEPATH=${cmake_dependency_file}.pre"
|
|
-D "output_file:FILEPATH=${cmake_dependency_file}.tmp"
|
|
-D "verbose=${verbose}"
|
|
-P "${HIP_run_make2cmake}"
|
|
)
|
|
|
|
if(HIP_result)
|
|
message(FATAL_ERROR "Error generating ${generated_file}")
|
|
endif()
|
|
|
|
# Copy the file if it is different
|
|
hip_execute_process(
|
|
"Copy if different ${cmake_dependency_file}.tmp to ${cmake_dependency_file}"
|
|
COMMAND "${CMAKE_COMMAND}" -E copy_if_different "${cmake_dependency_file}.tmp" "${cmake_dependency_file}"
|
|
)
|
|
|
|
if(HIP_result)
|
|
message(FATAL_ERROR "Error generating ${generated_file}")
|
|
endif()
|
|
|
|
# Delete the temporary file
|
|
hip_execute_process(
|
|
"Removing ${cmake_dependency_file}.tmp and ${cmake_dependency_file}.pre"
|
|
COMMAND "${CMAKE_COMMAND}" -E remove "${cmake_dependency_file}.tmp" "${cmake_dependency_file}.pre"
|
|
)
|
|
|
|
if(HIP_result)
|
|
message(FATAL_ERROR "Error generating ${generated_file}")
|
|
endif()
|
|
|
|
# Generate the output file
|
|
hip_execute_process(
|
|
"Generating ${generated_file}"
|
|
COMMAND "${__CC}"
|
|
-c
|
|
"${source_file}"
|
|
-o "${generated_file}"
|
|
${__CC_FLAGS}
|
|
${__CC_INCLUDES}
|
|
)
|
|
|
|
if(HIP_result)
|
|
# Make sure that we delete the output file
|
|
hip_execute_process(
|
|
"Removing ${generated_file}"
|
|
COMMAND "${CMAKE_COMMAND}" -E remove "${generated_file}"
|
|
)
|
|
message(FATAL_ERROR "Error generating file ${generated_file}")
|
|
else()
|
|
if(verbose)
|
|
message("Generated ${generated_file} successfully.")
|
|
endif()
|
|
endif()
|
|
# vim: ts=4:sw=4:expandtab:smartindent
|