From 0ea998134a1e5cada954e8be4932aba8d11ff067 Mon Sep 17 00:00:00 2001 From: Ryan Lai Date: Thu, 12 Nov 2020 18:08:11 -0800 Subject: [PATCH] Skip new x86 tests in ort model tests (#5789) --- onnxruntime/test/providers/cpu/model_tests.cc | 20 +++++++++++-------- 1 file changed, 12 insertions(+), 8 deletions(-) diff --git a/onnxruntime/test/providers/cpu/model_tests.cc b/onnxruntime/test/providers/cpu/model_tests.cc index c5e7fe4d25..a5360392e3 100644 --- a/onnxruntime/test/providers/cpu/model_tests.cc +++ b/onnxruntime/test/providers/cpu/model_tests.cc @@ -786,19 +786,23 @@ TEST_P(ModelTest, Run) { // This will be removed after LRU implementation all_disabled_tests.insert(std::begin(openvino_disabled_tests), std::end(openvino_disabled_tests)); } - + #if !defined(__amd64__) && !defined(_M_AMD64) // out of memory - static const ORTCHAR_T* x86_disabled_tests[] = {ORT_TSTR("mlperf_ssd_resnet34_1200"), + static const ORTCHAR_T* x86_disabled_tests[] = {ORT_TSTR("BERT_Squad"), + ORT_TSTR("bvlc_alexnet"), + ORT_TSTR("bvlc_reference_caffenet"), + ORT_TSTR("coreml_VGG16_ImageNet"), + ORT_TSTR("faster_rcnn"), + ORT_TSTR("GPT2"), + ORT_TSTR("GPT2_LM_HEAD"), + ORT_TSTR("keras_lotus_resnet3D"), + ORT_TSTR("mlperf_ssd_resnet34_1200"), ORT_TSTR("mask_rcnn_keras"), ORT_TSTR("mask_rcnn"), - ORT_TSTR("faster_rcnn"), - ORT_TSTR("vgg19"), - ORT_TSTR("zfnet512"), - ORT_TSTR("GPT2_LM_HEAD"), - ORT_TSTR("GPT2"), ORT_TSTR("ssd"), - ORT_TSTR("coreml_VGG16_ImageNet")}; + ORT_TSTR("vgg19"), + ORT_TSTR("zfnet512")}; all_disabled_tests.insert(std::begin(x86_disabled_tests), std::end(x86_disabled_tests)); #endif