[perftest] Enable perf tests for IREE EP

nod-ai · Oct 28, 2024 · 8eb88ec · 8eb88ec
1 parent ecebd73
commit 8eb88ec
Showing 1 changed file with 6 additions and 0 deletions.
diff --git 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