diff --git a/Common/ML/test/onnxruntime-inference/onnxruntime_ep_inference.cxx b/Common/ML/test/onnxruntime-inference/onnxruntime_ep_inference.cxx
index a066c193707a8..27be4f1643578 100644
--- a/Common/ML/test/onnxruntime-inference/onnxruntime_ep_inference.cxx
+++ b/Common/ML/test/onnxruntime-inference/onnxruntime_ep_inference.cxx
@@ -227,6 +227,7 @@ int main(int argc, char** argv)
     Ort::SessionOptions options;
     options.SetIntraOpNumThreads(1);
     options.SetGraphOptimizationLevel(GraphOptimizationLevel::ORT_ENABLE_ALL);
+    options.AddConfigEntry("session.record_ep_graph_assignment_info", "1");
     appendProvider(options, args);
 
     Ort::Session session(env, args.modelPath.c_str(), options);