aboutsummaryrefslogtreecommitdiffstats
path: root/host
diff options
context:
space:
mode:
authorMartin Braun <martin.braun@ettus.com>2015-01-26 12:17:17 +0100
committerMartin Braun <martin.braun@ettus.com>2015-10-19 17:33:17 -0700
commit508311768ad4f9538eecda16f1cae899ea31184b (patch)
tree5d244df138615d8fc055c07dc156cf72066af667 /host
parent4a6c47682daff6fae533647487297967ae559c49 (diff)
downloaduhd-508311768ad4f9538eecda16f1cae899ea31184b.tar.gz
uhd-508311768ad4f9538eecda16f1cae899ea31184b.tar.bz2
uhd-508311768ad4f9538eecda16f1cae899ea31184b.zip
tools: Added converter benchmark tool
Diffstat (limited to 'host')
-rw-r--r--host/utils/CMakeLists.txt11
-rw-r--r--host/utils/converter_benchmark.cpp432
-rw-r--r--host/utils/converter_benchmark.py193
3 files changed, 636 insertions, 0 deletions
diff --git a/host/utils/CMakeLists.txt b/host/utils/CMakeLists.txt
index 4f56dad0d..1ef7766f9 100644
--- a/host/utils/CMakeLists.txt
+++ b/host/utils/CMakeLists.txt
@@ -56,10 +56,14 @@ UHD_INSTALL(TARGETS usrp_x3xx_fpga_burner RUNTIME DESTINATION ${RUNTIME_DIR} COM
# Utilities that get installed into the share path
########################################################################
SET(util_share_sources
+ converter_benchmark.cpp
query_gpsdo_sensors.cpp
usrp_burn_db_eeprom.cpp
usrp_burn_mb_eeprom.cpp
)
+SET(util_share_sources_py
+ converter_benchmark.py
+)
IF(ENABLE_USB)
LIST(APPEND util_share_sources
fx2_init_eeprom.cpp
@@ -109,6 +113,13 @@ FOREACH(util_source ${util_share_sources})
TARGET_LINK_LIBRARIES(${util_name} uhd ${Boost_LIBRARIES})
UHD_INSTALL(TARGETS ${util_name} RUNTIME DESTINATION ${PKG_LIB_DIR}/utils COMPONENT utilities)
ENDFOREACH(util_source)
+FOREACH(util_source ${util_share_sources_py})
+ UHD_INSTALL(PROGRAMS
+ ${CMAKE_CURRENT_SOURCE_DIR}/${util_source}
+ DESTINATION ${PKG_LIB_DIR}/utils
+ COMPONENT utilities
+ )
+ENDFOREACH(util_source)
UHD_INSTALL(TARGETS usrp_n2xx_simple_net_burner RUNTIME DESTINATION ${PKG_LIB_DIR}/utils COMPONENT utilities)
UHD_INSTALL(TARGETS usrp_x3xx_fpga_burner RUNTIME DESTINATION ${PKG_LIB_DIR}/utils COMPONENT utilities)
diff --git a/host/utils/converter_benchmark.cpp b/host/utils/converter_benchmark.cpp
new file mode 100644
index 000000000..251dd6a37
--- /dev/null
+++ b/host/utils/converter_benchmark.cpp
@@ -0,0 +1,432 @@
+//
+// Copyright 2015 Ettus Research LLC
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+// GNU General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License
+// along with this program. If not, see <http://www.gnu.org/licenses/>.
+//
+
+#include <uhd/utils/safe_main.hpp>
+#include <uhd/types/dict.hpp>
+#include <uhd/convert.hpp>
+#include <uhd/exception.hpp>
+#include <boost/program_options.hpp>
+#include <boost/format.hpp>
+#include <boost/timer.hpp>
+#include <boost/algorithm/string.hpp>
+#include <iostream>
+#include <iomanip>
+#include <map>
+
+namespace po = boost::program_options;
+using namespace uhd::convert;
+
+enum buf_init_t {
+ RANDOM, INC
+};
+
+// Convert `sc16_item32_le' -> `sc16'
+// Finds the first _ in format and returns the string
+// until then. Returns the entire string if no _ is found.
+std::string format_to_type(const std::string &format)
+{
+ std::string ret_val = "";
+ for (size_t i = 0; i < format.length(); i++) {
+ if (format[i] == '_') {
+ return ret_val;
+ }
+ ret_val.append(1, format[i]);
+ }
+
+ return ret_val;
+}
+
+void configure_conv(
+ converter::sptr conv,
+ const std::string &in_type,
+ const std::string &out_type
+) {
+ if (in_type == "sc16") {
+ if (out_type == "fc32") {
+ std::cout << "Setting scalar to 32767." << std::endl;
+ conv->set_scalar(32767.);
+ return;
+ }
+ }
+
+ if (in_type == "fc32") {
+ if (out_type == "sc16") {
+ std::cout << "Setting scalar to 32767." << std::endl;
+ conv->set_scalar(32767.);
+ return;
+ }
+ }
+
+ std::cout << "No configuration required." << std::endl;
+}
+
+template <typename T>
+void init_random_vector_complex_float(std::vector<char> &buf_ptr, const size_t n_items)
+{
+ std::complex<T> * const buf = reinterpret_cast<std::complex<T> * const>(&buf_ptr[0]);
+ for (size_t i = 0; i < n_items; i++) {
+ buf[i] = std::complex<T>(
+ T((std::rand()/double(RAND_MAX/2)) - 1),
+ T((std::rand()/double(RAND_MAX/2)) - 1)
+ );
+ }
+}
+
+template <typename T>
+void init_random_vector_complex_int(std::vector<char> &buf_ptr, const size_t n_items)
+{
+ std::complex<T> * const buf = reinterpret_cast<std::complex<T> * const>(&buf_ptr[0]);
+ for (size_t i = 0; i < n_items; i++) {
+ buf[i] = std::complex<T>(T(std::rand()), T(std::rand()));
+ }
+}
+
+template <typename T>
+void init_random_vector_real_int(std::vector<char> &buf_ptr, size_t n_items)
+{
+ T * const buf = reinterpret_cast<T * const>(&buf_ptr[0]);
+ for (size_t i = 0; i < n_items; i++) {
+ buf[i] = T(std::rand());
+ }
+}
+
+// Fill a buffer with increasing numbers
+template <typename T>
+void init_inc_vector(std::vector<char> &buf_ptr, size_t n_items)
+{
+ T * const buf = reinterpret_cast<T * const>(&buf_ptr[0]);
+ for (size_t i = 0; i < n_items; i++) {
+ buf[i] = T(i);
+ }
+}
+
+void init_buffers(
+ std::vector< std::vector<char> > &buf,
+ const std::string &type,
+ size_t bytes_per_item,
+ buf_init_t buf_seed_mode
+) {
+ if (buf.empty()) {
+ return;
+ }
+ size_t n_items = buf[0].size() / bytes_per_item;
+
+ /// Fill with incrementing integers
+ if (buf_seed_mode == INC) {
+ for (size_t i = 0; i < buf.size(); i++) {
+ if (type == "sc8") {
+ init_inc_vector< std::complex<boost::int8_t> >(buf[i], n_items);
+ } else if (type == "sc16") {
+ init_inc_vector< std::complex<boost::int16_t> >(buf[i], n_items);
+ } else if (type == "sc32") {
+ init_inc_vector< std::complex<boost::int32_t> >(buf[i], n_items);
+ } else if (type == "fc32") {
+ init_inc_vector< std::complex<float> >(buf[i], n_items);
+ } else if (type == "fc64") {
+ init_inc_vector< std::complex<double> >(buf[i], n_items);
+ } else if (type == "s8") {
+ init_inc_vector< boost::int8_t >(buf[i], n_items);
+ } else if (type == "s16") {
+ init_inc_vector< boost::int16_t >(buf[i], n_items);
+ } else if (type == "item32") {
+ init_inc_vector< boost::uint32_t >(buf[i], n_items);
+ init_random_vector_real_int<boost::uint32_t>(buf[i], n_items);
+ } else {
+ throw uhd::runtime_error(str(
+ boost::format("Cannot handle data type: %s") % type
+ ));
+ }
+ }
+
+ return;
+ }
+
+ assert(buf_seed_mode == RANDOM);
+
+ /// Fill with random data
+ for (size_t i = 0; i < buf.size(); i++) {
+ if (type == "sc8") {
+ init_random_vector_complex_int<boost::int8_t>(buf[i], n_items);
+ } else if (type == "sc16") {
+ init_random_vector_complex_int<boost::int16_t>(buf[i], n_items);
+ } else if (type == "sc32") {
+ init_random_vector_complex_int<boost::int32_t>(buf[i], n_items);
+ } else if (type == "fc32") {
+ init_random_vector_complex_float<float>(buf[i], n_items);
+ } else if (type == "fc64") {
+ init_random_vector_complex_float<double>(buf[i], n_items);
+ } else if (type == "s8") {
+ init_random_vector_real_int<boost::int8_t>(buf[i], n_items);
+ } else if (type == "s16") {
+ init_random_vector_real_int<boost::int16_t>(buf[i], n_items);
+ } else if (type == "item32") {
+ init_random_vector_real_int<boost::uint32_t>(buf[i], n_items);
+ } else {
+ throw uhd::runtime_error(str(
+ boost::format("Cannot handle data type: %s") % type
+ ));
+ }
+ }
+}
+
+// Returns time elapsed
+double run_benchmark(
+ converter::sptr conv,
+ const std::vector<const void *> &input_buf_refs,
+ const std::vector<void *> &output_buf_refs,
+ size_t n_items,
+ size_t iterations
+) {
+ boost::timer benchmark_timer;
+ for (size_t i = 0; i < iterations; i++) {
+ conv->conv(input_buf_refs, output_buf_refs, n_items);
+ }
+ return benchmark_timer.elapsed();
+}
+
+template <typename T>
+std::string void_ptr_to_hexstring(const void *v_ptr, size_t index)
+{
+ const T *ptr = reinterpret_cast<const T *>(v_ptr);
+ return str(boost::format("%X") % ptr[index]);
+}
+
+std::string item_to_hexstring(
+ const void *v_ptr,
+ size_t index,
+ const std::string &type
+) {
+ if (type == "fc32") {
+ return void_ptr_to_hexstring<uint64_t>(v_ptr, index);
+ }
+ else if (type == "sc16" || type == "item32") {
+ return void_ptr_to_hexstring<uint32_t>(v_ptr, index);
+ }
+ else if (type == "sc8" || type == "s16") {
+ return void_ptr_to_hexstring<uint16_t>(v_ptr, index);
+ }
+ else if (type == "u8") {
+ return void_ptr_to_hexstring<uint8_t>(v_ptr, index);
+ }
+ else {
+ return str(boost::format("<unhandled data type: %s>") % type);
+ }
+}
+
+std::string item_to_string(
+ const void *v_ptr,
+ size_t index,
+ const std::string &type,
+ const bool print_hex
+) {
+ if (print_hex) {
+ return item_to_hexstring(v_ptr, index, type);
+ }
+
+ if (type == "sc16") {
+ const std::complex<boost::int16_t> *ptr = reinterpret_cast<const std::complex<boost::int16_t> *>(v_ptr);
+ return boost::lexical_cast<std::string>(ptr[index]);
+ }
+ else if (type == "sc8") {
+ const std::complex<boost::int8_t> *ptr = reinterpret_cast<const std::complex<boost::int8_t> *>(v_ptr);
+ return boost::lexical_cast<std::string>(ptr[index]);
+ }
+ else if (type == "fc32") {
+ const std::complex<float> *ptr = reinterpret_cast<const std::complex<float> *>(v_ptr);
+ return boost::lexical_cast<std::string>(ptr[index]);
+ }
+ else if (type == "item32") {
+ const boost::uint32_t *ptr = reinterpret_cast<const boost::uint32_t *>(v_ptr);
+ return boost::lexical_cast<std::string>(ptr[index]);
+ }
+ else if (type == "s16") {
+ const boost::int16_t *ptr = reinterpret_cast<const boost::int16_t *>(v_ptr);
+ return boost::lexical_cast<std::string>(ptr[index]);
+ }
+ else {
+ return str(boost::format("<unhandled data type: %s>") % type);
+ }
+}
+
+int UHD_SAFE_MAIN(int argc, char *argv[])
+{
+ std::string in_format, out_format;
+ std::string priorities;
+ std::string seed_mode;
+ priority_type prio = -1, max_prio;
+ size_t iterations, n_samples;
+ size_t n_inputs, n_outputs;
+ buf_init_t buf_seed_mode = RANDOM;
+
+ /// Command line arguments
+ po::options_description desc("Converter benchmark options:");
+ desc.add_options()
+ ("help", "help message")
+ ("in", po::value<std::string>(&in_format), "Input format (e.g. 'sc16')")
+ ("out", po::value<std::string>(&out_format), "Output format (e.g. 'sc16')")
+ ("samples", po::value<size_t>(&n_samples)->default_value(1000000), "Number of samples per iteration")
+ ("iterations", po::value<size_t>(&iterations)->default_value(10000), "Number of iterations per benchmark")
+ ("priorities", po::value<std::string>(&priorities)->default_value("default"), "Converter priorities. Can be 'default', 'all', or a comma-separated list of priorities.")
+ ("max-prio", po::value<priority_type>(&max_prio)->default_value(4), "Largest available priority (advanced feature)")
+ ("n-inputs", po::value<size_t>(&n_inputs)->default_value(1), "Number of input vectors")
+ ("n-outputs", po::value<size_t>(&n_outputs)->default_value(1), "Number of output vectors")
+ ("debug-converter", "Skip benchmark and print conversion results. Implies iterations==1 and will only run on a single converter.")
+ ("seed-mode", po::value<std::string>(&seed_mode)->default_value("random"), "How to initialize the data: random, incremental")
+ ("hex", "When using debug mode, dump memory in hex")
+ ;
+ po::variables_map vm;
+ po::store(po::parse_command_line(argc, argv, desc), vm);
+ po::notify(vm);
+
+ //print the help message
+ if (vm.count("help")){
+ std::cout << boost::format("UHD Converter Benchmark Tool %s") % desc << std::endl << std::endl;
+ std::cout << " Use this to benchmark or debug converters." << std::endl
+ << " When using as a benchmark tool, it will output the execution time\n"
+ " for every conversion run in CSV format to stdout. Every line between\n"
+ " the output delimiters {{{ }}} is of the format: <PRIO>,<TIME IN MILLISECONDS>\n"
+ " When using for converter debugging, every line is formatted as\n"
+ " <INPUT_VALUE>,<OUTPUT_VALUE>\n" << std::endl;
+ return EXIT_FAILURE;
+ }
+
+ // Parse more arguments
+ if (seed_mode == "incremental") {
+ buf_seed_mode = INC;
+ } else if (seed_mode == "random") {
+ buf_seed_mode = RANDOM;
+ } else {
+ std::cout << "Invalid argument: --seed-mode must be either 'incremental' or 'random'." << std::endl;
+ }
+
+ bool debug_mode = bool(vm.count("debug-converter"));
+ if (debug_mode) {
+ iterations = 1;
+ }
+
+ /// Create the converter(s) //////////////////////////////////////////////
+ id_type converter_id;
+ converter_id.input_format = in_format;
+ converter_id.output_format = out_format;
+ converter_id.num_inputs = n_inputs;
+ converter_id.num_outputs = n_outputs;
+ std::cout << "Requested converter format: " << converter_id.to_string()
+ << std::endl;
+ uhd::dict<priority_type, converter::sptr> conv_list;
+ if (priorities == "default" or priorities.empty()) {
+ try {
+ conv_list[prio] = get_converter(converter_id, prio)(); // Can throw a uhd::key_error
+ } catch(const uhd::key_error &e) {
+ std::cout << "No converters found." << std::endl;
+ return EXIT_FAILURE;
+ }
+ } else if (priorities == "all") {
+ for (priority_type i = 0; i < max_prio; i++) {
+ try {
+ // get_converter() returns a factory function, execute that immediately:
+ converter::sptr conv_for_prio = get_converter(converter_id, i)(); // Can throw a uhd::key_error
+ conv_list[i] = conv_for_prio;
+ } catch (...) {
+ continue;
+ }
+ }
+ } else { // Assume that priorities contains a list of prios (e.g. 0,2,3)
+ std::vector<std::string> prios_in_list;
+ boost::split(
+ prios_in_list,
+ priorities,
+ boost::is_any_of(","), // Split at ,
+ boost::token_compress_on // Avoid empty results
+ );
+ BOOST_FOREACH(const std::string &this_prio, prios_in_list) {
+ size_t prio_index = boost::lexical_cast<size_t>(this_prio);
+ converter::sptr conv_for_prio = get_converter(converter_id, prio_index)(); // Can throw a uhd::key_error
+ conv_list[prio_index] = conv_for_prio;
+ }
+ }
+ std::cout << "Found " << conv_list.size() << " converter(s)." << std::endl;
+
+ /// Create input and output buffers ///////////////////////////////////////
+ // First, convert the types to plain types (e.g. sc16_item32_le -> sc16)
+ const std::string in_type = format_to_type(in_format);
+ const std::string out_type = format_to_type(out_format);
+ const size_t in_size = get_bytes_per_item(in_type);
+ const size_t out_size = get_bytes_per_item(out_type);
+ // Create the buffers and fill them with random data & zeros, respectively
+ std::vector< std::vector<char> > input_buffers(n_inputs, std::vector<char>(in_size * n_samples, 0));
+ std::vector< std::vector<char> > output_buffers(n_outputs, std::vector<char>(out_size * n_samples, 0));
+ init_buffers(input_buffers, in_type, in_size, buf_seed_mode);
+ // Create ref vectors for the converter:
+ std::vector<const void *> input_buf_refs(n_inputs);
+ std::vector<void *> output_buf_refs(n_outputs);
+ for (size_t i = 0; i < n_inputs; i++) {
+ input_buf_refs[i] = reinterpret_cast<const void *>(&input_buffers[i][0]);
+ }
+ for (size_t i = 0; i < n_outputs; i++) {
+ output_buf_refs[i] = reinterpret_cast<void *>(&output_buffers[i][0]);
+ }
+
+ /// Final configurations to the converter:
+ std::cout << "Configuring converters:" << std::endl;
+ BOOST_FOREACH(priority_type prio_i, conv_list.keys()) {
+ std::cout << "* [" << prio_i << "]: ";
+ configure_conv(conv_list[prio_i], in_type, out_type);
+ }
+
+ /// Run the benchmark for every converter ////////////////////////////////
+ std::cout << "{{{" << std::endl;
+ if (not debug_mode) {
+ std::cout << "prio,duration_ms,avg_duration_ms,n_samples,iterations" << std::endl;
+ BOOST_FOREACH(priority_type prio_i, conv_list.keys()) {
+ double duration = run_benchmark(
+ conv_list[prio_i],
+ input_buf_refs,
+ output_buf_refs,
+ n_samples,
+ iterations
+ );
+ std::cout << boost::format("%i,%d,%d,%d,%d")
+ % prio_i
+ % (duration * 1000)
+ % (duration * 1000.0 / iterations)
+ % n_samples
+ % iterations
+ << std::endl;
+ }
+ }
+
+ /// Or run debug mode, which runs one conversion and prints the results ////
+ if (debug_mode) {
+ // Only run on the first converter:
+ run_benchmark(
+ conv_list[conv_list.keys().at(0)],
+ input_buf_refs,
+ output_buf_refs,
+ n_samples,
+ iterations
+ );
+ for (size_t i = 0; i < n_samples; i++) {
+ std::cout << item_to_string(input_buf_refs[0], i, in_type, vm.count("hex"))
+ << ";"
+ << item_to_string(reinterpret_cast< const void * >(output_buf_refs[0]), i, out_type, vm.count("hex"))
+ << std::endl;
+ }
+ }
+ std::cout << "}}}" << std::endl;
+
+ return EXIT_SUCCESS;
+}
diff --git a/host/utils/converter_benchmark.py b/host/utils/converter_benchmark.py
new file mode 100644
index 000000000..c3cab8753
--- /dev/null
+++ b/host/utils/converter_benchmark.py
@@ -0,0 +1,193 @@
+#!/usr/bin/env python
+#
+# Copyright 2015 Ettus Research LLC
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program. If not, see <http://www.gnu.org/licenses/>.
+#
+"""
+Wrap the converter_benchmark tool and produce prettier results.
+"""
+
+from __future__ import print_function
+import argparse
+import csv
+import subprocess
+
+INTRO_SETUP = {
+ 'n_samples': {
+ 'title': 'Samples per iteration',
+ },
+ 'iterations': {
+ 'title': 'Number of iterations'
+ },
+}
+
+TABLE_SETUP = {
+ 'prio': {
+ 'title': 'Priority',
+ },
+ 'duration_ms': {
+ 'title': 'Total Duration (ms)',
+ },
+ 'avg_duration_ms': {
+ 'title': 'Avg. Duration (ms)',
+ },
+}
+
+def run_benchmark(args):
+ """ Run the tool with the given arguments, return the section in the {{{ }}} brackets """
+ call_args = ['./converter_benchmark',]
+ for k, v in args.__dict__.iteritems():
+ k = k.replace('_', '-')
+ if v is None:
+ continue
+ if k in ('debug-converter', 'hex'):
+ if v:
+ call_args.append('--{0}'.format(k))
+ continue
+ call_args.append('--{0}'.format(k))
+ call_args.append(str(v))
+ print(call_args)
+ try:
+ output = subprocess.check_output(call_args)
+ except subprocess.CalledProcessError as ex:
+ print(ex.output)
+ exit(ex.returncode)
+ header_out, csv_output = output.split('{{{', 1)
+ csv_output = csv_output.split('}}}', 1)
+ assert len(csv_output) == 2 and csv_output[1].strip() == ''
+ return header_out, csv_output[0]
+
+def print_stats_table(args, csv_output):
+ """
+ Print stats.
+ """
+ reader = csv.reader(csv_output.strip().split('\n'), delimiter=',')
+ title_row = reader.next()
+ row_widths = [0,] * len(TABLE_SETUP)
+ for idx, row in enumerate(reader):
+ if idx == 0:
+ # Print intro:
+ for k, v in INTRO_SETUP.iteritems():
+ print("{title}: {value}".format(
+ title=v['title'],
+ value=row[title_row.index(k)],
+ ))
+ print("")
+ # Print table header
+ for idx, item in enumerate(TABLE_SETUP):
+ print(" {title} ".format(title=TABLE_SETUP[item]['title']), end='')
+ row_widths[idx] = len(TABLE_SETUP[item]['title'])
+ if idx < len(TABLE_SETUP) - 1:
+ print("|", end='')
+ print("")
+ for idx, item in enumerate(TABLE_SETUP):
+ print("-" * (row_widths[idx] + 2), end='')
+ if idx < len(TABLE_SETUP) - 1:
+ print("+", end='')
+ print("")
+ # Print actual row data
+ for idx, item in enumerate(TABLE_SETUP):
+ format_str = " {{item:>{n}}} ".format(n=row_widths[idx])
+ print(format_str.format(item=row[title_row.index(item)]), end='')
+ if idx < len(TABLE_SETUP) - 1:
+ print("|", end='')
+ print("")
+
+def print_debug_table(args, csv_output):
+ """
+ Print debug output.
+ """
+ reader = csv.reader(csv_output.strip().split('\n'), delimiter=';')
+ print_widths_hex = {
+ 'u8': 2,
+ 'sc16': 8,
+ 'fc32': 16,
+ 's16': 4,
+ }
+ if args.hex:
+ format_str = "{{0[0]:0>{n_in}}} => {{0[1]:0>{n_out}}}".format(
+ n_in=print_widths_hex[getattr(args, 'in').split('_', 1)[0]],
+ n_out=print_widths_hex[args.out.split('_', 1)[0]]
+ )
+ else:
+ format_str = "{0[0]}\t=>\t{0[1]}"
+ for row in reader:
+ print(format_str.format(row))
+
+def setup_argparse():
+ """ Configure arg parser. """
+ parser = argparse.ArgumentParser(
+ description="UHD Converter Benchmark + Debugging Utility.",
+ )
+ parser.add_argument(
+ "-i", "--in", required=True,
+ help="Input format (e.g. 'sc16')"
+ )
+ parser.add_argument(
+ "-o", "--out", required=True,
+ help="Output format (e.g. 'sc16')"
+ )
+ parser.add_argument(
+ "-s", "--samples", type=int,
+ help="Number of samples per iteration"
+ )
+ parser.add_argument(
+ "-N", "--iterations", type=int,
+ help="Number of iterations per benchmark",
+ )
+ parser.add_argument(
+ "-p", "--priorities",
+ help="Converter priorities. Can be 'default', 'all', or a comma-separated list of priorities.",
+ )
+ parser.add_argument(
+ "--max-prio", type=int,
+ help="Largest available priority (advanced feature)",
+ )
+ parser.add_argument(
+ "--n-inputs", type=int,
+ help="Number of input vectors",
+ )
+ parser.add_argument(
+ "--n-outputs", type=int,
+ help="Number of output vectors",
+ )
+ parser.add_argument(
+ "--seed-mode", choices=('random', 'incremental'),
+ help="How to initialize the data: random, incremental",
+ )
+ parser.add_argument(
+ "--debug-converter", action='store_true',
+ help="Skip benchmark and print conversion results. Implies iterations==1 and will only run on a single converter.",
+ )
+ parser.add_argument(
+ "--hex", action='store_true',
+ help="In debug mode, display data as hex values.",
+ )
+ return parser
+
+def main():
+ """ Go, go, go! """
+ args = setup_argparse().parse_args()
+ print("Running converter benchmark...")
+ header_out, csv_output = run_benchmark(args)
+ print(header_out)
+ if args.debug_converter:
+ print_debug_table(args, csv_output)
+ else:
+ print_stats_table(args, csv_output)
+
+if __name__ == "__main__":
+ main()
+