diff options
Diffstat (limited to 'tests/main.cpp')
-rw-r--r-- | tests/main.cpp | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/tests/main.cpp b/tests/main.cpp index 13f5aff14a..1e332cb457 100644 --- a/tests/main.cpp +++ b/tests/main.cpp @@ -84,6 +84,18 @@ bool file_exists(const std::string &filename) #endif /* ARM_COMPUTE_CL */ } //namespace +void print_cpu_info(std::ostream &os) +{ + const arm_compute::CPUInfo &cpu_info = Scheduler::get().cpu_info(); + const unsigned int num_cpus = cpu_info.get_cpu_num(); + os << "\ncpu_has_fp16 : " << cpu_info.has_fp16() << "\ncpu_has_dotprod : " << cpu_info.has_dotprod() << std::endl; + for(unsigned int j = 0; j < num_cpus; ++j) + { + const CPUModel model = cpu_info.get_cpu_model(j); + os << "CPU" << j << " : " << cpu_model_to_string(model) << std::endl; + } +} + int main(int argc, char **argv) { #ifdef ARM_COMPUTE_CL @@ -177,6 +189,7 @@ int main(int argc, char **argv) if(options.log_level->value() >= framework::LogLevel::CONFIG) { + std::stringstream ss; for(auto &p : printers) { p->print_entry("Version", build_information()); @@ -192,6 +205,8 @@ int main(int argc, char **argv) p->print_entry("CL_DEVICE_VERSION", "Unavailable"); } #endif /* ARM_COMPUTE_CL */ + print_cpu_info(ss); + p->print_entry("CPU_INFO", ss.str()); p->print_entry("Iterations", support::cpp11::to_string(options.iterations->value())); p->print_entry("Threads", support::cpp11::to_string(threads->value())); { |