diff --git a/tests/test_pipecache.cpp b/tests/test_pipecache.cpp index d83e831f5..b0818ddf4 100644 --- a/tests/test_pipecache.cpp +++ b/tests/test_pipecache.cpp @@ -65,9 +65,9 @@ static int warmup_gpu_pipecache() ncnn::Net net2; net2.opt.use_vulkan_compute = true; - net2.opt.pipeline_cache = new ncnn::PipelineCache(net.vulkan_device()); net2.load_param_mem("7767517\n2 2\nInput input0 0 1 input0\nSigmoid sigmoid0 1 1 input0 output0"); + net2.opt.pipeline_cache = new ncnn::PipelineCache(net2.vulkan_device()); if (net2.opt.pipeline_cache->load_cache(cache_path) != 0) { fprintf(stderr, "Warmup failed: could not load pipeline cache from %s\n", cache_path); @@ -162,18 +162,18 @@ static int test_gpu_pipecache_performance() printf("==================================================\n"); { ncnn::Net net_with_cache; - net_with_cache.opt.pipeline_cache = new ncnn::PipelineCache(ncnn::get_gpu_device()); net_with_cache.opt.use_vulkan_compute = true; auto start = ncnn::get_current_time(); + net_with_cache.load_param(MODEL_DIR "/mobilenet_v3.param"); // after load param vkdev will be create + net_with_cache.opt.pipeline_cache = new ncnn::PipelineCache(net_with_cache.vulkan_device()); // load from cache if (net_with_cache.opt.pipeline_cache->load_cache(cache_path) != 0) { fprintf(stderr, "Test failed: could not load pipeline cache from %s\n", cache_path); return -1; } - net_with_cache.load_param(MODEL_DIR "/mobilenet_v3.param"); net_with_cache.load_model(dr); auto end = ncnn::get_current_time();