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

Change-Id: I672b4a5287348bddf6eb8955decc9cce786275a2
Этот коммит содержится в:
Satyanvesh Dittakavi
2022-03-09 07:02:03 +00:00
родитель 16dbf85da0
Коммит af18c3144f
+3 -1
Просмотреть файл
@@ -34,6 +34,7 @@ THE SOFTWARE.
#define CUDA_10020 10020
#define CUDA_11010 11010
#define CUDA_11030 11030
#define CUDA_11040 11040
#ifdef __cplusplus
extern "C" {
@@ -2412,11 +2413,12 @@ inline static hipError_t hipGraphInstantiate(hipGraphExec_t* pGraphExec, hipGrap
cudaGraphInstantiate(pGraphExec, graph, pErrorNode, pLogBuffer, bufferSize));
}
#if CUDA_VERSION >= CUDA_11040
inline static hipError_t hipGraphInstantiateWithFlags(hipGraphExec_t* pGraphExec, hipGraph_t graph,
unsigned long long flags) {
return hipCUDAErrorTohipError(cudaGraphInstantiateWithFlags(pGraphExec, graph, flags));
}
#endif
inline static hipError_t hipGraphLaunch(hipGraphExec_t graphExec, hipStream_t stream) {
return hipCUDAErrorTohipError(cudaGraphLaunch(graphExec, stream));
}