From b2261e09c6d818f17cd0576ebd3a53a829be5e35 Mon Sep 17 00:00:00 2001 From: Leo Chen <39020268+leo0519@users.noreply.github.com> Date: Mon, 14 Aug 2023 18:05:52 +0800 Subject: [PATCH] Add WITH_MKLDNN check into target test_save_optimized_model_pass (#56113) --- test/ir/inference/CMakeLists.txt | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/test/ir/inference/CMakeLists.txt b/test/ir/inference/CMakeLists.txt index d010e954575..a3d4bc1f20c 100755 --- a/test/ir/inference/CMakeLists.txt +++ b/test/ir/inference/CMakeLists.txt @@ -174,7 +174,9 @@ if(WITH_GPU AND TENSORRT_FOUND) set_tests_properties(test_trt_inspector PROPERTIES TIMEOUT 60) set_tests_properties(test_trt_inference_predictor PROPERTIES TIMEOUT 60) set_tests_properties(test_trt_inference_fp16_io PROPERTIES TIMEOUT 300) - set_tests_properties(test_save_optimized_model_pass PROPERTIES TIMEOUT 300) + if(WITH_MKLDNN) + set_tests_properties(test_save_optimized_model_pass PROPERTIES TIMEOUT 300) + endif() if(WITH_NV_JETSON) set_tests_properties( -- GitLab