SWDEV-315118 - Add version check for hipGraphInstantiateWithFlags on nvidia path
Change-Id: I672b4a5287348bddf6eb8955decc9cce786275a2
Tá an tiomantas seo le fáil i:
@@ -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));
|
||||||
}
|
}
|
||||||
|
|||||||
Tagairt in Eagrán Nua
Cuir bac ar úsáideoir