Check CUDA_VERSION definition (openvinotoolkit#727)

apaniukov · Sep 15, 2023 · 57388ff · 57388ff
1 parent e2d59c1
commit 57388ff
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/modules/nvidia_plugin/src/cuda/graph.cpp b/modules/nvidia_plugin/src/cuda/graph.cpp
@@ -53,7 +53,7 @@ catch (std::exception &e) {
 }
 #endif
 
-#if CUDA_VERSION >= 12020
+#if defined(CUDA_VERSION) && CUDA_VERSION >= 12020
 cudaGraphExecUpdateResultInfo GraphExec::update(const Graph &g) const {
     cudaGraphExecUpdateResultInfo res;
     throwIfError(cudaGraphExecUpdate(get(), g.get(), &res));

diff --git a/modules/nvidia_plugin/src/cuda/graph.hpp b/modules/nvidia_plugin/src/cuda/graph.hpp
@@ -35,7 +35,7 @@ class GraphExec : public Handle<cudaGraphExec_t> {
 public:
     GraphExec(const Graph& g);
 
-#if CUDA_VERSION >= 12020
+#if defined(CUDA_VERSION) && CUDA_VERSION >= 12020
     cudaGraphExecUpdateResultInfo update(const Graph& g) const;
 #else
     cudaGraphExecUpdateResult update(const Graph& g) const;