From e6a674f7164bde674303e7fb3f1de22d7f5995f7 Mon Sep 17 00:00:00 2001
From: Giulio Eulisse <10544+ktf@users.noreply.github.com>
Date: Thu, 11 Apr 2024 13:18:51 +0200
Subject: [PATCH] Add support for ONNXRuntime 1.17.1

---
 Detectors/TRD/pid/include/TRDPID/ML.h                     | 4 ++++
 Detectors/TRD/pid/src/ML.cxx                              | 4 ++++
 Detectors/ZDC/fastsimulation/include/FastSimulations.h    | 4 ++++
 Detectors/ZDC/fastsimulation/include/Processors.h         | 4 ++++
 Detectors/ZDC/simulation/include/ZDCSimulation/Detector.h | 6 ++++++
 5 files changed, 22 insertions(+)

diff --git a/Detectors/TRD/pid/include/TRDPID/ML.h b/Detectors/TRD/pid/include/TRDPID/ML.h
index 5165996d24803..5b1248418e010 100644
--- a/Detectors/TRD/pid/include/TRDPID/ML.h
+++ b/Detectors/TRD/pid/include/TRDPID/ML.h
@@ -21,7 +21,11 @@
 #include "DataFormatsTRD/PID.h"
 #include "Framework/ProcessingContext.h"
 #include "Framework/InputRecord.h"
+#if __has_include (<onnxruntime/core/session/experimental_onnxruntime_cxx_api.h>)
 #include <onnxruntime/core/session/experimental_onnxruntime_cxx_api.h>
+#else
+#include <onnxruntime_cxx_api.h>
+#endif
 #include <memory>
 #include <vector>
 #include <array>
diff --git a/Detectors/TRD/pid/src/ML.cxx b/Detectors/TRD/pid/src/ML.cxx
index 488bf055a7f6c..2633e63820bb0 100644
--- a/Detectors/TRD/pid/src/ML.cxx
+++ b/Detectors/TRD/pid/src/ML.cxx
@@ -28,7 +28,11 @@
 #include "DetectorsBase/Propagator.h"
 
 #include <fmt/format.h>
+#if __has_include (<onnxruntime/core/session/experimental_onnxruntime_cxx_api.h>)
 #include <onnxruntime/core/session/experimental_onnxruntime_cxx_api.h>
+#else
+#include <onnxruntime_cxx_api.h>
+#endif
 #include <boost/range.hpp>
 
 #include <array>
diff --git a/Detectors/ZDC/fastsimulation/include/FastSimulations.h b/Detectors/ZDC/fastsimulation/include/FastSimulations.h
index 9244ceb894b0a..987b68efc8102 100644
--- a/Detectors/ZDC/fastsimulation/include/FastSimulations.h
+++ b/Detectors/ZDC/fastsimulation/include/FastSimulations.h
@@ -17,7 +17,11 @@
 #ifndef O2_ZDC_FAST_SIMULATIONS_H
 #define O2_ZDC_FAST_SIMULATIONS_H
 
+#if __has_include(<onnxruntime/core/sessions/onnxruntime_cxx_api.h>)
 #include <onnxruntime/core/session/onnxruntime_cxx_api.h>
+#else
+#include <onnxruntime_cxx_api.h>
+#endif
 #include <optional>
 #include <mutex>
 
diff --git a/Detectors/ZDC/fastsimulation/include/Processors.h b/Detectors/ZDC/fastsimulation/include/Processors.h
index b9836484fe871..1a0b4f609fc58 100644
--- a/Detectors/ZDC/fastsimulation/include/Processors.h
+++ b/Detectors/ZDC/fastsimulation/include/Processors.h
@@ -17,7 +17,11 @@
 #ifndef O2_ZDC_FAST_SIMULATIONS_PROCESSORS_H
 #define O2_ZDC_FAST_SIMULATIONS_PROCESSORS_H
 
+#if __has_include (<onnxruntime/core/session/onnxruntime_cxx_api.h>)
 #include <onnxruntime/core/session/onnxruntime_cxx_api.h>
+#else
+#include <onnxruntime_cxx_api.h>
+#endif
 #include <optional>
 #include <vector>
 
diff --git a/Detectors/ZDC/simulation/include/ZDCSimulation/Detector.h b/Detectors/ZDC/simulation/include/ZDCSimulation/Detector.h
index a150f8b78fe8b..6e0bf0667a95b 100644
--- a/Detectors/ZDC/simulation/include/ZDCSimulation/Detector.h
+++ b/Detectors/ZDC/simulation/include/ZDCSimulation/Detector.h
@@ -26,7 +26,13 @@
 
 // inclusions and forward decl for fast sim
 #ifdef ZDC_FASTSIM_ONNX
+
+#if __has_include(<onnxruntime/core/session/onnxruntime_cxx_api.h>)
 #include <onnxruntime/core/session/onnxruntime_cxx_api.h>
+#else
+#include <onnxruntime_cxx_api.h>
+#endif
+
 namespace o2::zdc
 {
 namespace fastsim