diff options
Diffstat (limited to 'src/core/NEON/kernels/arm_gemm/profiler.hpp')
-rw-r--r-- | src/core/NEON/kernels/arm_gemm/profiler.hpp | 133 |
1 files changed, 133 insertions, 0 deletions
diff --git a/src/core/NEON/kernels/arm_gemm/profiler.hpp b/src/core/NEON/kernels/arm_gemm/profiler.hpp new file mode 100644 index 0000000000..c38b0a443c --- /dev/null +++ b/src/core/NEON/kernels/arm_gemm/profiler.hpp @@ -0,0 +1,133 @@ +/* + * Copyright (c) 2017-2018 ARM Limited. + * + * SPDX-License-Identifier: MIT + * + * Permission is hereby granted, free of charge, to any person obtaining a copy + * of this software and associated documentation files (the "Software"), to + * deal in the Software without restriction, including without limitation the + * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or + * sell copies of the Software, and to permit persons to whom the Software is + * furnished to do so, subject to the following conditions: + * + * The above copyright notice and this permission notice shall be included in all + * copies or substantial portions of the Software. + * + * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR + * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, + * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE + * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER + * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, + * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE + * SOFTWARE. + */ +#pragma once + +#ifdef CYCLE_PROFILING + +#include "../perf.h" + +#ifndef NO_MULTI_THREADING +#include <mutex> +#endif + +namespace arm_gemm +{ +#ifndef NO_MULTI_THREADING +extern std::mutex report_mutex; +#endif + +class profiler +{ +private: + static const int maxevents = 100000; + unsigned long times[maxevents] = {}; + unsigned long units[maxevents] = {}; + int events[maxevents] = {}; + int currentevent = 0; + int countfd = 0; + +public: + profiler() + { + countfd = open_cycle_counter(); + } + + ~profiler() + { + close(countfd); + int tots[5]; + unsigned long counts[5]; + unsigned long tunits[5]; + const char *descs[] = { "Prepare A", "Prepare B", "Kernel", "Merge" }; + + for(int i = 1; i < 5; i++) + { + tots[i] = 0; + counts[i] = 0; + tunits[i] = 0; + } + + for(int i = 0; i < currentevent; i++) + { + // printf("%10s: %ld\n", descs[events[i]-1], times[i]); + tots[events[i]]++; + counts[events[i]] += times[i]; + tunits[events[i]] += units[i]; + } + +#ifdef NO_MULTI_THREADING + printf("Profiled events:\n"); +#else + std::lock_guard<std::mutex> lock(report_mutex); + printf("Profiled events (cpu %d):\n", sched_getcpu()); +#endif + + printf("%20s %9s %9s %9s %12s %9s\n", "", "Events", "Total", "Average", "Bytes/MACs", "Per cycle"); + for(int i = 1; i < 5; i++) + { + printf("%20s: %9d %9ld %9ld %12lu %9.2f\n", descs[i - 1], tots[i], counts[i], counts[i] / tots[i], tunits[i], (float)tunits[i] / counts[i]); + } + } + + template <typename T> + void operator()(int i, unsigned long u, T func) + { + if(currentevent == maxevents) + { + func(); + } + else + { + events[currentevent] = i; + units[currentevent] = u; + start_counter(countfd); + func(); + long long cycs = stop_counter(countfd); + times[currentevent++] = cycs; + } + } +}; + +#else + +namespace arm_gemm +{ +class profiler +{ +public: + template <typename T> + void operator()(int i, unsigned long u, T func) + { + func(); + } +}; + +#endif // CYCLE_PROFILING + +} // namespace arm_gemm + +#define PROFILE_PREPA 1 +#define PROFILE_PREPB 2 +#define PROFILE_KERNEL 3 +#define PROFILE_MERGE 4 |