ROCm CI Service Account
363 строки
11 KiB
C
363 строки
11 KiB
C
/*
|
|
Copyright (c) 2015 - 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.
|
|
*/
|
|
#pragma once
|
|
|
|
#include <hip/hip_common.h>
|
|
|
|
#if !(defined(__HIP_PLATFORM_HCC__) || defined(__HIP_PLATFORM_AMD__)) && (defined(__HIP_PLATFORM_NVCC__) || defined(__HIP_PLATFORM_NVIDIA__))
|
|
#include <hip/nvidia_detail/nvidia_hiprtc.h>
|
|
#elif (defined(__HIP_PLATFORM_HCC__) || defined(__HIP_PLATFORM_AMD__)) && !(defined(__HIP_PLATFORM_NVCC__) || defined(__HIP_PLATFORM_NVIDIA__))
|
|
|
|
/**
|
|
* @addtogroup Runtime Runtime Compilation
|
|
* @{
|
|
* @ingroup Runtime
|
|
*/
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif /* __cplusplus */
|
|
|
|
#include <stdlib.h>
|
|
|
|
#if !defined(_WIN32)
|
|
#pragma GCC visibility push (default)
|
|
#endif
|
|
|
|
/**
|
|
* @brief hiprtcResult
|
|
* @enum
|
|
*
|
|
*/
|
|
|
|
typedef enum hiprtcResult {
|
|
HIPRTC_SUCCESS = 0,
|
|
HIPRTC_ERROR_OUT_OF_MEMORY = 1,
|
|
HIPRTC_ERROR_PROGRAM_CREATION_FAILURE = 2,
|
|
HIPRTC_ERROR_INVALID_INPUT = 3,
|
|
HIPRTC_ERROR_INVALID_PROGRAM = 4,
|
|
HIPRTC_ERROR_INVALID_OPTION = 5,
|
|
HIPRTC_ERROR_COMPILATION = 6,
|
|
HIPRTC_ERROR_LINKING = 7,
|
|
HIPRTC_ERROR_BUILTIN_OPERATION_FAILURE = 8,
|
|
HIPRTC_ERROR_NO_NAME_EXPRESSIONS_AFTER_COMPILATION = 9,
|
|
HIPRTC_ERROR_NO_LOWERED_NAMES_BEFORE_COMPILATION = 10,
|
|
HIPRTC_ERROR_NAME_EXPRESSION_NOT_VALID = 11,
|
|
HIPRTC_ERROR_INTERNAL_ERROR = 12
|
|
} hiprtcResult;
|
|
|
|
typedef enum hiprtcJIT_option {
|
|
HIPRTC_JIT_MAX_REGISTERS = 0,
|
|
HIPRTC_JIT_THREADS_PER_BLOCK,
|
|
HIPRTC_JIT_WALL_TIME,
|
|
HIPRTC_JIT_INFO_LOG_BUFFER,
|
|
HIPRTC_JIT_INFO_LOG_BUFFER_SIZE_BYTES,
|
|
HIPRTC_JIT_ERROR_LOG_BUFFER,
|
|
HIPRTC_JIT_ERROR_LOG_BUFFER_SIZE_BYTES,
|
|
HIPRTC_JIT_OPTIMIZATION_LEVEL,
|
|
HIPRTC_JIT_TARGET_FROM_HIPCONTEXT,
|
|
HIPRTC_JIT_TARGET,
|
|
HIPRTC_JIT_FALLBACK_STRATEGY,
|
|
HIPRTC_JIT_GENERATE_DEBUG_INFO,
|
|
HIPRTC_JIT_LOG_VERBOSE,
|
|
HIPRTC_JIT_GENERATE_LINE_INFO,
|
|
HIPRTC_JIT_CACHE_MODE,
|
|
HIPRTC_JIT_NEW_SM3X_OPT,
|
|
HIPRTC_JIT_FAST_COMPILE,
|
|
HIPRTC_JIT_GLOBAL_SYMBOL_NAMES,
|
|
HIPRTC_JIT_GLOBAL_SYMBOL_ADDRESS,
|
|
HIPRTC_JIT_GLOBAL_SYMBOL_COUNT,
|
|
HIPRTC_JIT_LTO,
|
|
HIPRTC_JIT_FTZ,
|
|
HIPRTC_JIT_PREC_DIV,
|
|
HIPRTC_JIT_PREC_SQRT,
|
|
HIPRTC_JIT_FMA,
|
|
HIPRTC_JIT_NUM_OPTIONS,
|
|
} hiprtcJIT_option;
|
|
|
|
typedef enum hiprtcJITInputType {
|
|
HIPRTC_JIT_INPUT_CUBIN = 0,
|
|
HIPRTC_JIT_INPUT_PTX,
|
|
HIPRTC_JIT_INPUT_FATBINARY,
|
|
HIPRTC_JIT_INPUT_OBJECT,
|
|
HIPRTC_JIT_INPUT_LIBRARY,
|
|
HIPRTC_JIT_INPUT_NVVM,
|
|
HIPRTC_JIT_NUM_LEGACY_INPUT_TYPES,
|
|
HIPRTC_JIT_INPUT_LLVM_BITCODE = 100,
|
|
HIPRTC_JIT_INPUT_LLVM_BUNDLED_BITCODE = 101,
|
|
HIPRTC_JIT_INPUT_LLVM_ARCHIVES_OF_BUNDLED_BITCODE = 102,
|
|
HIPRTC_JIT_NUM_INPUT_TYPES = (HIPRTC_JIT_NUM_LEGACY_INPUT_TYPES + 3)
|
|
} hiprtcJITInputType;
|
|
|
|
typedef struct ihiprtcLinkState* hiprtcLinkState;
|
|
|
|
/**
|
|
* @brief Returns text string message to explain the error which occurred
|
|
*
|
|
* @param [in] result code to convert to string.
|
|
* @return const char pointer to the NULL-terminated error string
|
|
*
|
|
* @warning In HIP, this function returns the name of the error,
|
|
* if the hiprtc result is defined, it will return "Invalid HIPRTC error code"
|
|
*
|
|
* @see hiprtcResult
|
|
*/
|
|
const char* hiprtcGetErrorString(hiprtcResult result);
|
|
|
|
/**
|
|
* @brief Sets the parameters as major and minor version.
|
|
*
|
|
* @param [out] major HIP Runtime Compilation major version.
|
|
* @param [out] minor HIP Runtime Compilation minor version.
|
|
*
|
|
*/
|
|
hiprtcResult hiprtcVersion(int* major, int* minor);
|
|
|
|
typedef struct _hiprtcProgram* hiprtcProgram;
|
|
|
|
/**
|
|
* @brief Adds the given name exprssion to the runtime compilation program.
|
|
*
|
|
* @param [in] prog runtime compilation program instance.
|
|
* @param [in] name_expression const char pointer to the name expression.
|
|
* @return HIPRTC_SUCCESS
|
|
*
|
|
* If const char pointer is NULL, it will return HIPRTC_ERROR_INVALID_INPUT.
|
|
*
|
|
* @see hiprtcResult
|
|
*/
|
|
hiprtcResult hiprtcAddNameExpression(hiprtcProgram prog,
|
|
const char* name_expression);
|
|
|
|
/**
|
|
* @brief Compiles the given runtime compilation program.
|
|
*
|
|
* @param [in] prog runtime compilation program instance.
|
|
* @param [in] numOptions number of compiler options.
|
|
* @param [in] options compiler options as const array of strins.
|
|
* @return HIPRTC_SUCCESS
|
|
*
|
|
* If the compiler failed to build the runtime compilation program,
|
|
* it will return HIPRTC_ERROR_COMPILATION.
|
|
*
|
|
* @see hiprtcResult
|
|
*/
|
|
hiprtcResult hiprtcCompileProgram(hiprtcProgram prog,
|
|
int numOptions,
|
|
const char** options);
|
|
|
|
/**
|
|
* @brief Creates an instance of hiprtcProgram with the given input parameters,
|
|
* and sets the output hiprtcProgram prog with it.
|
|
*
|
|
* @param [in, out] prog runtime compilation program instance.
|
|
* @param [in] src const char pointer to the program source.
|
|
* @param [in] name const char pointer to the program name.
|
|
* @param [in] numHeaders number of headers.
|
|
* @param [in] headers array of strings pointing to headers.
|
|
* @param [in] includeNames array of strings pointing to names included in program source.
|
|
* @return HIPRTC_SUCCESS
|
|
*
|
|
* Any invalide input parameter, it will return HIPRTC_ERROR_INVALID_INPUT
|
|
* or HIPRTC_ERROR_INVALID_PROGRAM.
|
|
*
|
|
* If failed to create the program, it will return HIPRTC_ERROR_PROGRAM_CREATION_FAILURE.
|
|
*
|
|
* @see hiprtcResult
|
|
*/
|
|
hiprtcResult hiprtcCreateProgram(hiprtcProgram* prog,
|
|
const char* src,
|
|
const char* name,
|
|
int numHeaders,
|
|
const char** headers,
|
|
const char** includeNames);
|
|
|
|
/**
|
|
* @brief Destroys an instance of given hiprtcProgram.
|
|
*
|
|
* @param [in] prog runtime compilation program instance.
|
|
* @return HIPRTC_SUCCESS
|
|
*
|
|
* If prog is NULL, it will return HIPRTC_ERROR_INVALID_INPUT.
|
|
*
|
|
* @see hiprtcResult
|
|
*/
|
|
hiprtcResult hiprtcDestroyProgram(hiprtcProgram* prog);
|
|
|
|
/**
|
|
* @brief Gets the lowered (mangled) name from an instance of hiprtcProgram with the given input parameters,
|
|
* and sets the output lowered_name with it.
|
|
*
|
|
* @param [in] prog runtime compilation program instance.
|
|
* @param [in] name_expression const char pointer to the name expression.
|
|
* @param [in, out] lowered_name const char array to the lowered (mangled) name.
|
|
* @return HIPRTC_SUCCESS
|
|
*
|
|
* If any invalide nullptr input parameters, it will return HIPRTC_ERROR_INVALID_INPUT
|
|
*
|
|
* If name_expression is not found, it will return HIPRTC_ERROR_NAME_EXPRESSION_NOT_VALID
|
|
*
|
|
* If failed to get lowered_name from the program, it will return HIPRTC_ERROR_COMPILATION.
|
|
*
|
|
* @see hiprtcResult
|
|
*/
|
|
hiprtcResult hiprtcGetLoweredName(hiprtcProgram prog,
|
|
const char* name_expression,
|
|
const char** lowered_name);
|
|
|
|
/**
|
|
* @brief Gets the log generated by the runtime compilation program instance.
|
|
*
|
|
* @param [in] prog runtime compilation program instance.
|
|
* @param [out] log memory pointer to the generated log.
|
|
* @return HIPRTC_SUCCESS
|
|
*
|
|
* @see hiprtcResult
|
|
*/
|
|
hiprtcResult hiprtcGetProgramLog(hiprtcProgram prog, char* log);
|
|
|
|
/**
|
|
* @brief Gets the size of log generated by the runtime compilation program instance.
|
|
*
|
|
* @param [in] prog runtime compilation program instance.
|
|
* @param [out] logSizeRet size of generated log.
|
|
* @return HIPRTC_SUCCESS
|
|
*
|
|
* @see hiprtcResult
|
|
*/
|
|
hiprtcResult hiprtcGetProgramLogSize(hiprtcProgram prog,
|
|
size_t* logSizeRet);
|
|
|
|
/**
|
|
* @brief Gets the pointer of compilation binary by the runtime compilation program instance.
|
|
*
|
|
* @param [in] prog runtime compilation program instance.
|
|
* @param [out] code char pointer to binary.
|
|
* @return HIPRTC_SUCCESS
|
|
*
|
|
* @see hiprtcResult
|
|
*/
|
|
hiprtcResult hiprtcGetCode(hiprtcProgram prog, char* code);
|
|
|
|
/**
|
|
* @brief Gets the size of compilation binary by the runtime compilation program instance.
|
|
*
|
|
* @param [in] prog runtime compilation program instance.
|
|
* @param [out] code the size of binary.
|
|
* @return HIPRTC_SUCCESS
|
|
*
|
|
* @see hiprtcResult
|
|
*/
|
|
hiprtcResult hiprtcGetCodeSize(hiprtcProgram prog, size_t* codeSizeRet);
|
|
|
|
/**
|
|
* @brief Creates the link instance via hiprtc APIs.
|
|
*
|
|
* @param [in] hip_jit_options
|
|
* @param [out] hiprtc link state instance
|
|
* @return HIPRTC_SUCCESS
|
|
*
|
|
* @see hiprtcResult
|
|
*/
|
|
hiprtcResult hiprtcLinkCreate(unsigned int num_options, hiprtcJIT_option* option_ptr,
|
|
void** option_vals_pptr, hiprtcLinkState* hip_link_state_ptr);
|
|
|
|
/**
|
|
* @brief Adds a file with bit code to be linked with options
|
|
*
|
|
* @param [in] hiprtc link state, jit input type, file path,
|
|
* option reated parameters.
|
|
* @param [out] None.
|
|
* @return HIPRTC_SUCCESS
|
|
*
|
|
* If input values are invalid, it will
|
|
* @return HIPRTC_ERROR_INVALID_INPUT
|
|
*
|
|
* @see hiprtcResult
|
|
*/
|
|
|
|
hiprtcResult hiprtcLinkAddFile(hiprtcLinkState hip_link_state, hiprtcJITInputType input_type,
|
|
const char* file_path, unsigned int num_options,
|
|
hiprtcJIT_option* options_ptr, void** option_values);
|
|
|
|
/**
|
|
* @brief Completes the linking of the given program.
|
|
*
|
|
* @param [in] hiprtc link state, jit input type, image_ptr ,
|
|
* option reated parameters.
|
|
* @param [out] None.
|
|
* @return HIPRTC_SUCCESS
|
|
*
|
|
* If adding the file fails, it will
|
|
* @return HIPRTC_ERROR_PROGRAM_CREATION_FAILURE
|
|
*
|
|
* @see hiprtcResult
|
|
*/
|
|
|
|
hiprtcResult hiprtcLinkAddData(hiprtcLinkState hip_link_state, hiprtcJITInputType input_type,
|
|
void* image, size_t image_size, const char* name,
|
|
unsigned int num_options, hiprtcJIT_option* options_ptr,
|
|
void** option_values);
|
|
|
|
/**
|
|
* @brief Completes the linking of the given program.
|
|
*
|
|
* @param [in] hiprtc link state instance
|
|
* @param [out] linked_binary, linked_binary_size.
|
|
* @return HIPRTC_SUCCESS
|
|
*
|
|
* If adding the data fails, it will
|
|
* @return HIPRTC_ERROR_PROGRAM_CREATION_FAILURE
|
|
*
|
|
* @see hiprtcResult
|
|
*/
|
|
hiprtcResult hiprtcLinkComplete(hiprtcLinkState hip_link_state, void** bin_out, size_t* size_out);
|
|
|
|
/**
|
|
* @brief Deletes the link instance via hiprtc APIs.
|
|
*
|
|
* @param [in] hiprtc link state instance
|
|
* @param [out] code the size of binary.
|
|
* @return HIPRTC_SUCCESS
|
|
*
|
|
* If linking fails, it will
|
|
* @return HIPRTC_ERROR_LINKING
|
|
*
|
|
* @see hiprtcResult
|
|
*/
|
|
hiprtcResult hiprtcLinkDestroy(hiprtcLinkState hip_link_state);
|
|
|
|
#if !defined(_WIN32)
|
|
#pragma GCC visibility pop
|
|
#endif
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif /* __cplusplus */
|
|
// doxygen end HIPrtc feature
|
|
/**
|
|
* @}
|
|
*/
|
|
#else
|
|
#error("Must define exactly one of __HIP_PLATFORM_AMD__ or __HIP_PLATFORM_NVIDIA__");
|
|
#endif
|