From 8eb88ec78679f37c5fa75618440ffc57e59e5a3e Mon Sep 17 00:00:00 2001 From: Vinayak Dev Date: Mon, 28 Oct 2024 18:27:29 +0530 Subject: [PATCH] [perftest] Enable perf tests for IREE EP --- onnxruntime/test/perftest/ort_test_session.cc | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/onnxruntime/test/perftest/ort_test_session.cc b/onnxruntime/test/perftest/ort_test_session.cc index 38496893d2c05..ef53214107bd6 100644 --- a/onnxruntime/test/perftest/ort_test_session.cc +++ b/onnxruntime/test/perftest/ort_test_session.cc @@ -398,6 +398,12 @@ select from 'TF8', 'TF16', 'UINT8', 'FLOAT', 'ITENSOR'. \n)"); Ort::ThrowOnError(OrtSessionOptionsAppendExecutionProvider_Nnapi(session_options, nnapi_flags)); #else ORT_THROW("NNAPI is not supported in this build\n"); +#endif + } else if (provider_name_ == onnxruntime::kIreeExecutionProvider) { +#ifdef USE_IREE + session_options.AppendExecutionProvider("IREE"); +#else + ORT_THROW("IREE is not supported in this build\n"); #endif } else if (provider_name_ == onnxruntime::kVSINPUExecutionProvider) { #ifdef USE_VSINPU