diff options
Diffstat (limited to 'tests/benchmark/DragonBenchConfigs.h')
-rw-r--r-- | tests/benchmark/DragonBenchConfigs.h | 54 |
1 files changed, 53 insertions, 1 deletions
diff --git a/tests/benchmark/DragonBenchConfigs.h b/tests/benchmark/DragonBenchConfigs.h index c12a22c8ae..4fe6c79ae8 100644 --- a/tests/benchmark/DragonBenchConfigs.h +++ b/tests/benchmark/DragonBenchConfigs.h @@ -28,10 +28,59 @@ #include <string> #include <vector> -#include "dragonbench/conv2d.hpp" +#include "dragonbench/conv2d/conv2d.hpp" +#include "dragonbench/fully_connected/fully_connected.hpp" namespace arm_compute { +// Stream operators +inline ::std::ostream &operator<<(::std::ostream &os, const Conv2D &conv2d_config) +{ + os << "network_name=" << conv2d_config.network_name << ":"; + os << "layer_name=" << conv2d_config.layer_name << ":"; + os << "id=" << conv2d_config.id << ":"; + os << "Input_NCHW=" + << conv2d_config.ibatch << ',' + << conv2d_config.ch_in << ',' + << conv2d_config.dim_in_h << ',' + << conv2d_config.dim_in_w << ":"; + os << "Output_NCHW=" + << conv2d_config.ibatch << ',' + << conv2d_config.ch_out << ',' + << conv2d_config.dim_out_h << ',' + << conv2d_config.dim_out_w << ":"; + os << "Weights_HW=" + << conv2d_config.kern_h << ',' + << conv2d_config.kern_w << ":"; + os << "Stride_HW=" + << conv2d_config.stride_h << ',' + << conv2d_config.stride_w << ":"; + os << "Padding=" << conv2d_config.padding << ":"; + return os; +} +inline std::string to_string(const Conv2D &conv2d_config) +{ + std::stringstream str; + str << conv2d_config; + return str.str(); +} +inline ::std::ostream &operator<<(::std::ostream &os, const Fully_Connected &fc_config) +{ + os << "network_name=" << fc_config.network_name << ":"; + os << "layer_name=" << fc_config.layer_name << ":"; + os << "id=" << fc_config.id << ":"; + os << "M=" << fc_config.m << ":"; + os << "N=" << fc_config.n << ":"; + os << "K=" << fc_config.k << ":"; + return os; +} +inline std::string to_string(const Fully_Connected &fc_config) +{ + std::stringstream str; + str << fc_config; + return str.str(); +} + namespace test { namespace benchmark @@ -40,6 +89,9 @@ namespace benchmark extern conv2d_configs silverwing_cfgs; extern conv2d_configs sunfyre_cfgs; extern conv2d_configs syrax_cfgs; + +// Fully Connected benchmarks +extern fully_connected_configs dreamfyre_cfgs; } // namespace benchmark } // namespace test } // namespace arm_compute |