From e4f71abd90fbb2e7c5e2ec350eba9c9d8eda18c4 Mon Sep 17 00:00:00 2001 From: Changming Sun Date: Thu, 1 Oct 2020 21:49:45 -0700 Subject: [PATCH] Exclude GPT2_LM_HEAD from OpenVino's model test list (#5356) GPT2_LM_HEAD is a new ONNX model zoo model that OpenVino doesn't support. Error message:1: [ONNXRuntimeError] : 6 : RUNTIME_EXCEPTION : Non-zero status code returned while running OpenVINO-EP-subgraph_1162 node. Name:'OpenVINOExecutionProvider_OpenVINO-EP-subgraph_1162_1' Status Message: _Map_base::at --- onnxruntime/test/providers/cpu/model_tests.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/onnxruntime/test/providers/cpu/model_tests.cc b/onnxruntime/test/providers/cpu/model_tests.cc index f0191f0623..b825f00943 100644 --- a/onnxruntime/test/providers/cpu/model_tests.cc +++ b/onnxruntime/test/providers/cpu/model_tests.cc @@ -676,6 +676,7 @@ TEST_P(ModelTest, Run) { ORT_TSTR("mlperf_ssd_resnet34_1200"), ORT_TSTR("candy"), ORT_TSTR("cntk_simple_seg"), + ORT_TSTR("GPT2_LM_HEAD"), ORT_TSTR("negative_log_likelihood_loss_input_shape_is_NCd1d2d3d4d5_mean_weight"), ORT_TSTR("negative_log_likelihood_loss_input_shape_is_NCd1d2d3d4d5_mean_weight_expanded"), ORT_TSTR("negative_log_likelihood_loss_input_shape_is_NCd1d2d3d4d5_none_no_weight"),