SWDEV-282361 - HIP support for NVRTC
Change-Id: I82869af8194b595f9a2fb64ad034630786189335
[ROCm/hip commit: 4de1bfafcc]
Este commit está contenido en:
Anusha GodavarthySurya
@@ -26,7 +26,7 @@ THE SOFTWARE.
|
||||
#if (defined(__HIP_PLATFORM_HCC__) || defined(__HIP_PLATFORM_AMD__)) && !(defined(__HIP_PLATFORM_NVCC__) || defined(__HIP_PLATFORM_NVIDIA__))
|
||||
#include <hip/amd_detail/hiprtc.h>
|
||||
#elif !(defined(__HIP_PLATFORM_HCC__) || defined(__HIP_PLATFORM_AMD__)) && (defined(__HIP_PLATFORM_NVCC__) || defined(__HIP_PLATFORM_NVIDIA__))
|
||||
#include <hip/nvidia_detail/nvrtc.h>
|
||||
#include <hip/nvidia_detail/hiprtc.h>
|
||||
#else
|
||||
#error("Must define exactly one of __HIP_PLATFORM_AMD__ or __HIP_PLATFORM_NVIDIA__");
|
||||
#endif
|
||||
|
||||
@@ -0,0 +1,168 @@
|
||||
/*
|
||||
Copyright (c) 2021 - present 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.
|
||||
*/
|
||||
#ifndef HIPRTC_H
|
||||
#define HIPRTC_H
|
||||
|
||||
#include <cuda.h>
|
||||
#include <nvrtc.h>
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif /* __cplusplus */
|
||||
|
||||
#include <stdlib.h>
|
||||
|
||||
#if !defined(_WIN32)
|
||||
#pragma GCC visibility push(default)
|
||||
#endif
|
||||
|
||||
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_BUILTIN_OPERATION_FAILURE = 7,
|
||||
HIPRTC_ERROR_NO_NAME_EXPRESSIONS_AFTER_COMPILATION = 8,
|
||||
HIPRTC_ERROR_NO_LOWERED_NAMES_BEFORE_COMPILATION = 9,
|
||||
HIPRTC_ERROR_NAME_EXPRESSION_NOT_VALID = 10,
|
||||
HIPRTC_ERROR_INTERNAL_ERROR = 11
|
||||
} hiprtcResult;
|
||||
|
||||
inline static nvrtcResult hiprtcResultTonvrtcResult(hiprtcResult result) {
|
||||
switch (result) {
|
||||
case HIPRTC_SUCCESS:
|
||||
return NVRTC_SUCCESS;
|
||||
case HIPRTC_ERROR_OUT_OF_MEMORY:
|
||||
return NVRTC_ERROR_OUT_OF_MEMORY;
|
||||
case HIPRTC_ERROR_PROGRAM_CREATION_FAILURE:
|
||||
return NVRTC_ERROR_PROGRAM_CREATION_FAILURE;
|
||||
case HIPRTC_ERROR_INVALID_INPUT:
|
||||
return NVRTC_ERROR_INVALID_INPUT;
|
||||
case HIPRTC_ERROR_INVALID_PROGRAM:
|
||||
return NVRTC_ERROR_INVALID_PROGRAM;
|
||||
case HIPRTC_ERROR_INVALID_OPTION:
|
||||
return NVRTC_ERROR_INVALID_OPTION;
|
||||
case HIPRTC_ERROR_COMPILATION:
|
||||
return NVRTC_ERROR_COMPILATION;
|
||||
case HIPRTC_ERROR_BUILTIN_OPERATION_FAILURE:
|
||||
return NVRTC_ERROR_BUILTIN_OPERATION_FAILURE;
|
||||
case HIPRTC_ERROR_NO_NAME_EXPRESSIONS_AFTER_COMPILATION:
|
||||
return NVRTC_ERROR_NO_NAME_EXPRESSIONS_AFTER_COMPILATION;
|
||||
case HIPRTC_ERROR_NO_LOWERED_NAMES_BEFORE_COMPILATION:
|
||||
return NVRTC_ERROR_NO_LOWERED_NAMES_BEFORE_COMPILATION;
|
||||
case HIPRTC_ERROR_NAME_EXPRESSION_NOT_VALID:
|
||||
return NVRTC_ERROR_NAME_EXPRESSION_NOT_VALID;
|
||||
case HIPRTC_ERROR_INTERNAL_ERROR:
|
||||
return NVRTC_ERROR_INTERNAL_ERROR;
|
||||
}
|
||||
}
|
||||
|
||||
inline static hiprtcResult nvrtcResultTohiprtcResult(nvrtcResult result) {
|
||||
switch (result) {
|
||||
case NVRTC_SUCCESS:
|
||||
return HIPRTC_SUCCESS;
|
||||
case NVRTC_ERROR_OUT_OF_MEMORY:
|
||||
return HIPRTC_ERROR_OUT_OF_MEMORY;
|
||||
case NVRTC_ERROR_PROGRAM_CREATION_FAILURE:
|
||||
return HIPRTC_ERROR_PROGRAM_CREATION_FAILURE;
|
||||
case NVRTC_ERROR_INVALID_INPUT:
|
||||
return HIPRTC_ERROR_INVALID_INPUT;
|
||||
case NVRTC_ERROR_INVALID_PROGRAM:
|
||||
return HIPRTC_ERROR_INVALID_PROGRAM;
|
||||
case NVRTC_ERROR_INVALID_OPTION:
|
||||
return HIPRTC_ERROR_INVALID_OPTION;
|
||||
case NVRTC_ERROR_COMPILATION:
|
||||
return HIPRTC_ERROR_COMPILATION;
|
||||
case NVRTC_ERROR_BUILTIN_OPERATION_FAILURE:
|
||||
return HIPRTC_ERROR_BUILTIN_OPERATION_FAILURE;
|
||||
case NVRTC_ERROR_NO_NAME_EXPRESSIONS_AFTER_COMPILATION:
|
||||
return HIPRTC_ERROR_NO_NAME_EXPRESSIONS_AFTER_COMPILATION;
|
||||
case NVRTC_ERROR_NO_LOWERED_NAMES_BEFORE_COMPILATION:
|
||||
return HIPRTC_ERROR_NO_LOWERED_NAMES_BEFORE_COMPILATION;
|
||||
case NVRTC_ERROR_NAME_EXPRESSION_NOT_VALID:
|
||||
return HIPRTC_ERROR_NAME_EXPRESSION_NOT_VALID;
|
||||
case NVRTC_ERROR_INTERNAL_ERROR:
|
||||
return HIPRTC_ERROR_INTERNAL_ERROR;
|
||||
}
|
||||
}
|
||||
|
||||
const char* hiprtcGetErrorString(hiprtcResult result) {
|
||||
return nvrtcGetErrorString(hiprtcResultTonvrtcResult(result));
|
||||
}
|
||||
|
||||
hiprtcResult hiprtcVersion(int* major, int* minor) {
|
||||
return nvrtcResultTohiprtcResult(nvrtcVersion(major, minor));
|
||||
}
|
||||
|
||||
typedef nvrtcProgram hiprtcProgram;
|
||||
|
||||
hiprtcResult hiprtcAddNameExpression(hiprtcProgram prog, const char* name_expression) {
|
||||
return nvrtcResultTohiprtcResult(nvrtcAddNameExpression(prog, name_expression));
|
||||
}
|
||||
|
||||
hiprtcResult hiprtcCompileProgram(hiprtcProgram prog, int numOptions, const char** options) {
|
||||
return nvrtcResultTohiprtcResult(nvrtcCompileProgram(prog, numOptions, options));
|
||||
}
|
||||
|
||||
hiprtcResult hiprtcCreateProgram(hiprtcProgram* prog, const char* src, const char* name,
|
||||
int numHeaders, const char** headers, const char** includeNames) {
|
||||
return nvrtcResultTohiprtcResult(
|
||||
nvrtcCreateProgram(prog, src, name, numHeaders, headers, includeNames));
|
||||
}
|
||||
|
||||
hiprtcResult hiprtcDestroyProgram(hiprtcProgram* prog) {
|
||||
return nvrtcResultTohiprtcResult(nvrtcDestroyProgram(prog));
|
||||
}
|
||||
|
||||
hiprtcResult hiprtcGetLoweredName(hiprtcProgram prog, const char* name_expression,
|
||||
const char** lowered_name) {
|
||||
return nvrtcResultTohiprtcResult(nvrtcGetLoweredName(prog, name_expression, lowered_name));
|
||||
}
|
||||
|
||||
hiprtcResult hiprtcGetProgramLog(hiprtcProgram prog, char* log) {
|
||||
return nvrtcResultTohiprtcResult(nvrtcGetProgramLog(prog, log));
|
||||
}
|
||||
|
||||
hiprtcResult hiprtcGetProgramLogSize(hiprtcProgram prog, size_t* logSizeRet) {
|
||||
return nvrtcResultTohiprtcResult(nvrtcGetProgramLogSize(prog, logSizeRet));
|
||||
}
|
||||
|
||||
hiprtcResult hiprtcGetCode(hiprtcProgram prog, char* code) {
|
||||
return nvrtcResultTohiprtcResult(nvrtcGetPTX(prog, code));
|
||||
}
|
||||
|
||||
hiprtcResult hiprtcGetCodeSize(hiprtcProgram prog, size_t* codeSizeRet) {
|
||||
return nvrtcResultTohiprtcResult(nvrtcGetPTXSize(prog, codeSizeRet));
|
||||
}
|
||||
|
||||
#if !defined(_WIN32)
|
||||
#pragma GCC visibility pop
|
||||
#endif
|
||||
|
||||
#ifdef __cplusplus
|
||||
}
|
||||
#endif /* __cplusplus */
|
||||
|
||||
#endif // HIPRTC_H
|
||||
Referencia en una nueva incidencia
Block a user