diff --git a/mindspore/lite/CMakeLists.txt b/mindspore/lite/CMakeLists.txt index 76d2674eb8..5d13dc68fc 100644 --- a/mindspore/lite/CMakeLists.txt +++ b/mindspore/lite/CMakeLists.txt @@ -132,7 +132,7 @@ if(SUPPORT_GPU STREQUAL vulkan) include(${TOP_DIR}/cmake/external_libs/vulkan.cmake) endif() -if(ENABLE_CONVERTER OR BUILD_MINDDATA STREQUAL "full" OR BUILD_MINDDATA STREQUAL "wrapper" OR PLATFORM_ARM) +if(ENABLE_CONVERTER OR BUILD_MINDDATA STREQUAL "full" OR BUILD_MINDDATA STREQUAL "wrapper" OR ENABLE_TOOLS) include(${TOP_DIR}/cmake/external_libs/json.cmake) endif() diff --git a/mindspore/lite/tools/benchmark/benchmark.cc b/mindspore/lite/tools/benchmark/benchmark.cc index 214bdba9d7..a074fe2d94 100644 --- a/mindspore/lite/tools/benchmark/benchmark.cc +++ b/mindspore/lite/tools/benchmark/benchmark.cc @@ -1050,8 +1050,18 @@ int Benchmark::InitDumpConfigFromJson(char *path) { } } - dump_file_output_dir_ = dump_cfg_json_[dump::kSettings][dump::kPath].get() + "/" + - dump_cfg_json_[dump::kSettings][dump::kNetName].get(); + auto abs_path = dump_cfg_json_[dump::kSettings][dump::kPath].get(); + auto net_name = dump_cfg_json_[dump::kSettings][dump::kNetName].get(); + if (abs_path.back() == '\\' || abs_path.back() == '/') { + dump_file_output_dir_ = abs_path + net_name; + } else { +#ifdef _WIN32 + dump_file_output_dir_ = abs_path + "\\" + net_name; +#else + dump_file_output_dir_ = abs_path + "/" + net_name; +#endif + } + auto status = CreateOutputDir(&dump_file_output_dir_); if (status != RET_OK) { MS_LOG(ERROR) << "create data output directory failed.";