diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2021-06-09 10:08:27 +0100 |
---|---|---|
committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2021-06-15 17:57:39 +0000 |
commit | 08302c17cd57356b35d46e17dc8d8f76672da5cf (patch) | |
tree | 3fed9bea3586bc140cc953e935f6ed55e8692dac /tests | |
parent | 450dfb1b4d719d60295bfae56f4c46dcaf044d72 (diff) | |
download | ComputeLibrary-08302c17cd57356b35d46e17dc8d8f76672da5cf.tar.gz |
Add CPU discovery capabilities.
Resolves: COMPMID-4500
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Change-Id: I008c51934ef813fb1f489b531288c4419e701955
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5799
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'tests')
-rw-r--r-- | tests/main.cpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/tests/main.cpp b/tests/main.cpp index 308338e781..e1963b1609 100644 --- a/tests/main.cpp +++ b/tests/main.cpp @@ -45,6 +45,7 @@ #include "utils/TypePrinter.h" #endif /* ARM_COMPUTE_CL */ #include "arm_compute/runtime/Scheduler.h" +#include "src/common/cpuinfo/CpuModel.h" #include <fstream> #include <initializer_list> @@ -235,13 +236,15 @@ int main(int argc, char **argv) #endif /* ARM_COMPUTE_CL */ const arm_compute::CPUInfo &cpu_info = Scheduler::get().cpu_info(); const unsigned int num_cpus = cpu_info.get_cpu_num(); + p->print_entry("cpu_has_sve", support::cpp11::to_string(cpu_info.has_sve())); p->print_entry("cpu_has_fp16", support::cpp11::to_string(cpu_info.has_fp16())); + p->print_entry("cpu_has_bf16", support::cpp11::to_string(cpu_info.has_bf16())); p->print_entry("cpu_has_dotprod", support::cpp11::to_string(cpu_info.has_dotprod())); for(unsigned int j = 0; j < num_cpus; ++j) { const CPUModel model = cpu_info.get_cpu_model(j); - p->print_entry("CPU" + support::cpp11::to_string(j), cpu_model_to_string(model)); + p->print_entry("CPU" + support::cpp11::to_string(j), cpuinfo::cpu_model_to_string(model)); } p->print_entry("Iterations", support::cpp11::to_string(options.iterations->value())); p->print_entry("Threads", support::cpp11::to_string(threads->value())); |