[BENCHMARK_APP] Show input command in output (#13402)

This commit is contained in:
Piotr Krzemiński
2022-10-26 10:58:09 +02:00
committed by GitHub
parent 848d880eb3
commit f893a58816
2 changed files with 44 additions and 0 deletions

View File

@@ -28,10 +28,41 @@
#include "remote_tensors_filling.hpp"
#include "statistics_report.hpp"
#include "utils.hpp"
#if defined(_WIN32) || defined(WIN32)
# include <windows.h>
#endif
// clang-format on
static const size_t progressBarDefaultTotalCount = 1000;
std::string get_console_command(int argc, char* argv[]) {
std::stringstream args_command;
#if defined(_WIN32) || defined(WIN32)
std::string relative_path(argv[0]);
std::vector<char> buffer;
uint32_t len = 1024;
do {
buffer.resize(len);
len = GetFullPathNameA(relative_path.data(), len, buffer.data(), nullptr);
} while (len > buffer.size());
std::string full_path(buffer.begin(), buffer.end());
args_command << full_path;
#else
args_command << realpath(argv[0], nullptr);
#endif
args_command << " ";
for (int i = 1; i < argc; i++) {
args_command << argv[i] << " ";
}
return args_command.str();
}
bool parse_and_check_command_line(int argc, char* argv[]) {
// ---------------------------Parsing and validating input
// arguments--------------------------------------
@@ -167,12 +198,19 @@ int main(int argc, char* argv[]) {
// ----------------- 1. Parsing and validating input arguments
// -------------------------------------------------
// Must be executed before parse_and_check_command_line()
// gflags::ParseCommandLineNonHelpFlags() modifies the argv array
auto command_from_args = get_console_command(argc, argv);
next_step();
if (!parse_and_check_command_line(argc, argv)) {
return 0;
}
slog::info << "Input command: " << command_from_args << slog::endl;
bool isNetworkCompiled = fileExt(FLAGS_m) == "blob";
if (isNetworkCompiled) {
slog::info << "Network is compiled" << slog::endl;