Skip new x86 tests in ort model tests (#5789)

This commit is contained in:
Ryan Lai 2020-11-12 18:08:11 -08:00 committed by GitHub
parent 2f35e65135
commit 0ea998134a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -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