Skip to content

Commit

Permalink
Disable zfnet512 test for x86
Browse files Browse the repository at this point in the history
  • Loading branch information
RyanUnderhill committed Jan 23, 2020
1 parent 1a5a253 commit 2cec09a
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions onnxruntime/test/onnx/main.cc
Original file line number Diff line number Diff line change
Expand Up @@ -360,7 +360,7 @@ int real_main(int argc, char* argv[], Ort::Env& env) {
"test_resnet101v2", "test_vgg19", "tf_inception_resnet_v2", "tf_inception_v1", "tf_inception_v3", "tf_inception_v4", "tf_mobilenet_v1_1.0_224",
"tf_mobilenet_v2_1.0_224", "tf_mobilenet_v2_1.4_224", "tf_nasnet_large", "tf_pnasnet_large", "tf_resnet_v1_50", "tf_resnet_v1_101", "tf_resnet_v1_101",
"tf_resnet_v2_101", "tf_resnet_v2_152"};

std::unordered_set<std::string> all_disabled_tests;
if (enable_cuda) {
all_disabled_tests.insert(std::begin(cuda_flaky_tests), std::end(cuda_flaky_tests));
Expand All @@ -376,7 +376,7 @@ int real_main(int argc, char* argv[], Ort::Env& env) {
#if defined(__amd64__) || defined(_M_AMD64)
#else
//out of memory
static const char* x86_disabled_tests[] = {"mlperf_ssd_resnet34_1200", "mask_rcnn_keras", "mask_rcnn", "faster_rcnn", "vgg19"};
static const char* x86_disabled_tests[] = {"mlperf_ssd_resnet34_1200", "mask_rcnn_keras", "mask_rcnn", "faster_rcnn", "vgg19", "zfnet512"};
all_disabled_tests.insert(std::begin(x86_disabled_tests), std::end(x86_disabled_tests));
#endif

Expand Down

0 comments on commit 2cec09a

Please sign in to comment.