diff --git a/Control/AthOnnxruntimeUtils/AthOnnxruntimeUtils/OnnxUtils.h b/Control/AthOnnxruntimeUtils/AthOnnxruntimeUtils/OnnxUtils.h index 21e1096a360a68912d13a3fe38299b8feda44cd7..cc4816226daba766d1d73977d6f0d0df69d206ac 100644 --- a/Control/AthOnnxruntimeUtils/AthOnnxruntimeUtils/OnnxUtils.h +++ b/Control/AthOnnxruntimeUtils/AthOnnxruntimeUtils/OnnxUtils.h @@ -67,7 +67,7 @@ namespace AthONNX { Ort::SessionOptions sessionOptions; sessionOptions.SetIntraOpNumThreads( 1 ); if (withCUDA) { - Ort::ThrowOnError(OrtSessionOptionsAppendExecutionProvider_CUDA(sessionOptions, 0)); + ; // does nothing for now until we have a GPU enabled build } sessionOptions.SetGraphOptimizationLevel( GraphOptimizationLevel::ORT_ENABLE_EXTENDED ); @@ -80,28 +80,6 @@ namespace AthONNX { } -/****************************Creation of ORT Session for GPU******************************/ -/*****************************************************************************************/ - - //template<typename T> - inline std::unique_ptr< Ort::Session > CreateORTSessionGPU(const std::string& modelFile){ - - // Set up the ONNX Runtime session. - Ort::SessionOptions sessionOptions; - Ort::ThrowOnError (OrtSessionOptionsAppendExecutionProvider_CUDA(sessionOptions,0)); - sessionOptions.SetIntraOpNumThreads( 1 ); - sessionOptions.SetGraphOptimizationLevel( ORT_ENABLE_BASIC ); - - ServiceHandle< IONNXRuntimeSvc > svc("AthONNX::ONNXRuntimeSvc", - "AthONNX::ONNXRuntimeSvc"); - - return std::make_unique<Ort::Session>( svc->env(), - modelFile.c_str(), - sessionOptions ); - } - - - /*********************************Input Node Structure of Model*********************************/ /***********************************************************************************************/ diff --git a/Control/AthOnnxruntimeUtils/CMakeLists.txt b/Control/AthOnnxruntimeUtils/CMakeLists.txt index 42e610f5cac2849698f32714d34be088b862f9bb..de53f529136561a42e24f320d920b35ef9b9c65d 100644 --- a/Control/AthOnnxruntimeUtils/CMakeLists.txt +++ b/Control/AthOnnxruntimeUtils/CMakeLists.txt @@ -9,4 +9,3 @@ atlas_add_library( AthOnnxruntimeUtilsLib PUBLIC_HEADERS AthOnnxruntimeUtils INCLUDE_DIRS ${ONNXRUNTIME_INCLUDE_DIRS} LINK_LIBRARIES ${ONNXRUNTIME_LIBRARIES} AthOnnxruntimeServiceLib AthenaKernel GaudiKernel AsgServicesLib ) -