diff --git a/paddle/fluid/jit/CMakeLists.txt b/paddle/fluid/jit/CMakeLists.txt index 40d2dd307fb1f703f6acd3cd4289faa4e212a732..43f63945ae5a4d18bca943faa986f5c863e71013 100644 --- a/paddle/fluid/jit/CMakeLists.txt +++ b/paddle/fluid/jit/CMakeLists.txt @@ -13,15 +13,13 @@ cc_library( SRCS base_function.cc DEPS scope proto_desc) -if(WITH_TESTING AND NOT WIN32) +if(WITH_TESTING + AND NOT WIN32 + AND NOT "$ENV{CI_SKIP_CPP_TEST}" STREQUAL "ON") add_custom_target( jit_download_program - COMMAND wget -nc --no-proxy - https://paddle-ci.gz.bcebos.com/dy2st/export.forward.pdiparams - COMMAND wget -nc --no-proxy - https://paddle-ci.gz.bcebos.com/dy2st/export.forward.pdmodel - COMMAND wget -nc --no-proxy - https://paddle-ci.gz.bcebos.com/dy2st/export.infer.pdmodel) + COMMAND wget -nc https://paddle-ci.gz.bcebos.com/dy2st/Testing.tar.gz + COMMAND tar zxvf Testing.tar.gz) set(JIT_DEPS phi elementwise_add_op diff --git a/paddle/fluid/jit/layer_test.cc b/paddle/fluid/jit/layer_test.cc index 6f8b7a58b842225a07664a2efe57a530f8663c84..dfd46818bde1fa98ac49b26440accba1c6b216a5 100644 --- a/paddle/fluid/jit/layer_test.cc +++ b/paddle/fluid/jit/layer_test.cc @@ -76,7 +76,7 @@ TEST(CpuLayerTest, Construct) { paddle::imperative::SetCurrentTracer(tracer); imperative::GetCurrentTracer()->SetExpectedPlace(phi::CPUPlace()); - std::string path = "./"; + std::string path = "./Testing/"; auto layer = jit::Load(path); auto inputs = PrepareInputs(); @@ -105,7 +105,7 @@ TEST(GpuLayerTest, Construct) { const auto* dev_ctx_gpu = static_cast(&dev_ctx); DenseTensor cpu_dense_tensor; - std::string path = "./"; + std::string path = "./Testing/"; auto layer = jit::Load(path); auto inputs = PrepareInputs();