From 00d525f5a27e9712f6701cdd985d24f68cd70133 Mon Sep 17 00:00:00 2001
From: Zlobin Vladimir <vladimir.zlobin@intel.com>
Date: Wed, 5 Jun 2024 22:43:26 +0400
Subject: [PATCH] Export OpenVINOGenAI_DIR (#24683) (#24870)

Co-authored-by: Ilya Lavrenov <ilya.lavrenov@intel.com>
---
 scripts/setupvars/setupvars.bat | 4 ++++
 scripts/setupvars/setupvars.ps1 | 5 +++++
 scripts/setupvars/setupvars.sh  | 2 ++
 3 files changed, 11 insertions(+)

diff --git a/scripts/setupvars/setupvars.bat b/scripts/setupvars/setupvars.bat
index 65f11e58ba2..1e401696382 100644
--- a/scripts/setupvars/setupvars.bat
+++ b/scripts/setupvars/setupvars.bat
@@ -34,6 +34,10 @@ if exist "%INTEL_OPENVINO_DIR%\extras\opencv\setupvars.bat" (
 
 :: OpenVINO runtime
 set "OpenVINO_DIR=%INTEL_OPENVINO_DIR%\runtime\cmake"
+if exist "%OpenVINO_DIR%\OpenVINOGenAIConfig.cmake" (
+   :: If GenAI is installed, export it as well.
+   set OpenVINOGenAI_DIR=%OpenVINO_DIR%
+)
 set "OPENVINO_LIB_PATHS=%INTEL_OPENVINO_DIR%\runtime\bin\intel64\Release;%INTEL_OPENVINO_DIR%\runtime\bin\intel64\Debug;%OPENVINO_LIB_PATHS%"
 
 :: TBB
diff --git a/scripts/setupvars/setupvars.ps1 b/scripts/setupvars/setupvars.ps1
index f64869b0455..c7e20473548 100644
--- a/scripts/setupvars/setupvars.ps1
+++ b/scripts/setupvars/setupvars.ps1
@@ -9,6 +9,11 @@ param (
 $Env:INTEL_OPENVINO_DIR = Split-Path $MyInvocation.MyCommand.Path
 
 $Env:OpenVINO_DIR = "$Env:INTEL_OPENVINO_DIR/runtime/cmake"
+if (Test-Path -Path "$Env:OpenVINO_DIR/OpenVINOGenAIConfig.cmake")
+{
+    # If GenAI is installed, export it as well.
+    $Env:OpenVINOGenAI_DIR = $Env:OpenVINO_DIR
+}
 $Env:OPENVINO_LIB_PATHS = "$Env:INTEL_OPENVINO_DIR/runtime/bin/intel64/Release;$Env:INTEL_OPENVINO_DIR/runtime/bin/intel64/Debug;$Env:OPENVINO_LIB_PATHS"
 
 # TBB
diff --git a/scripts/setupvars/setupvars.sh b/scripts/setupvars/setupvars.sh
index e2dc114a5a2..54eefc25714 100755
--- a/scripts/setupvars/setupvars.sh
+++ b/scripts/setupvars/setupvars.sh
@@ -33,6 +33,8 @@ done
 
 if [ -e "$INSTALLDIR/runtime" ]; then
     export OpenVINO_DIR=$INSTALLDIR/runtime/cmake
+    # If GenAI is installed, export it as well.
+    [ -f "$OpenVINO_DIR/OpenVINOGenAIConfig.cmake" ] && export OpenVINOGenAI_DIR=$OpenVINO_DIR
 
     system_type=$(/bin/ls "$INSTALLDIR/runtime/lib/")
     OV_PLUGINS_PATH=$INSTALLDIR/runtime/lib/$system_type