SWDEV-315118 - Add version check for hipGraphInstantiateWithFlags on nvidia path

Change-Id: I672b4a5287348bddf6eb8955decc9cce786275a2
Tá an tiomantas seo le fáil i:
Satyanvesh Dittakavi
2022-03-09 07:02:03 +00:00
tuismitheoir 16dbf85da0
tiomantas af18c3144f
@@ -34,6 +34,7 @@ THE SOFTWARE.
#define CUDA_10020 10020 #define CUDA_10020 10020
#define CUDA_11010 11010 #define CUDA_11010 11010
#define CUDA_11030 11030 #define CUDA_11030 11030
#define CUDA_11040 11040
#ifdef __cplusplus #ifdef __cplusplus
extern "C" { extern "C" {
@@ -2412,11 +2413,12 @@ inline static hipError_t hipGraphInstantiate(hipGraphExec_t* pGraphExec, hipGrap
cudaGraphInstantiate(pGraphExec, graph, pErrorNode, pLogBuffer, bufferSize)); cudaGraphInstantiate(pGraphExec, graph, pErrorNode, pLogBuffer, bufferSize));
} }
#if CUDA_VERSION >= CUDA_11040
inline static hipError_t hipGraphInstantiateWithFlags(hipGraphExec_t* pGraphExec, hipGraph_t graph, inline static hipError_t hipGraphInstantiateWithFlags(hipGraphExec_t* pGraphExec, hipGraph_t graph,
unsigned long long flags) { unsigned long long flags) {
return hipCUDAErrorTohipError(cudaGraphInstantiateWithFlags(pGraphExec, graph, flags)); return hipCUDAErrorTohipError(cudaGraphInstantiateWithFlags(pGraphExec, graph, flags));
} }
#endif
inline static hipError_t hipGraphLaunch(hipGraphExec_t graphExec, hipStream_t stream) { inline static hipError_t hipGraphLaunch(hipGraphExec_t graphExec, hipStream_t stream) {
return hipCUDAErrorTohipError(cudaGraphLaunch(graphExec, stream)); return hipCUDAErrorTohipError(cudaGraphLaunch(graphExec, stream));
} }