Skip to content

Commit 42273fb

Browse files
committed
[perftest] Enable perf tests for IREE EP
1 parent ecebd73 commit 42273fb

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

onnxruntime/test/perftest/ort_test_session.cc

+6
Original file line numberDiff line numberDiff line change
@@ -398,6 +398,12 @@ select from 'TF8', 'TF16', 'UINT8', 'FLOAT', 'ITENSOR'. \n)");
398398
Ort::ThrowOnError(OrtSessionOptionsAppendExecutionProvider_Nnapi(session_options, nnapi_flags));
399399
#else
400400
ORT_THROW("NNAPI is not supported in this build\n");
401+
#endif
402+
} else if (provider_name_ == onnxruntime::kIreeExecutionProvider) {
403+
#ifdef USE_IREE
404+
session_options.AppendExecutionProvider("IREE");
405+
#else
406+
ORT_THROW("IREE is not supported in this build\n");
401407
#endif
402408
} else if (provider_name_ == onnxruntime::kVSINPUExecutionProvider) {
403409
#ifdef USE_VSINPU

0 commit comments

Comments
 (0)