diff options
Diffstat (limited to 'runtimes/contrib/xtrace/src/xtrace.cc')
-rw-r--r-- | runtimes/contrib/xtrace/src/xtrace.cc | 64 |
1 files changed, 64 insertions, 0 deletions
diff --git a/runtimes/contrib/xtrace/src/xtrace.cc b/runtimes/contrib/xtrace/src/xtrace.cc new file mode 100644 index 000000000..117a2e663 --- /dev/null +++ b/runtimes/contrib/xtrace/src/xtrace.cc @@ -0,0 +1,64 @@ +/* + * Copyright (c) 2019 Samsung Electronics Co., Ltd. All Rights Reserved + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +#include "event_recorder.h" +#include "event_collector.h" +#include "benchmark_runner.h" + +#include <cassert> +#include <fstream> +#include <iostream> +#include <string> + +// xtrace --out <output path> <T/F Lite model path) +static int entry(int argc, char **argv) +{ + assert(argc == 4); + assert(std::string(argv[1]) == "--out"); + + // Create a file + std::ofstream ofs{argv[2], std::ofstream::out}; + + // Create an event recorder + EventRecorder recorder{ofs}; + + recorder.init(); + + EventCollector event_collector{&recorder}; + + xray::mux::get().attach(&event_collector); + + BMRunner<TFL_NNAPI_DELEGATE>().run(argv[3]); + + xray::mux::get().detach(&event_collector); + + recorder.fini(); + + return 0; +} + +int main(int argc, char **argv) +{ + try + { + return entry(argc, argv); + } + catch (const std::exception &e) + { + std::cerr << e.what() << std::endl; + return 255; + } +} |