-
Notifications
You must be signed in to change notification settings - Fork 326
/
f32-rdsum.cc
117 lines (94 loc) · 3.63 KB
/
f32-rdsum.cc
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
// Copyright 2024 Google LLC
//
// This source code is licensed under the BSD-style license found in the
// LICENSE file in the root directory of this source tree.
//
// Auto-generated file. Do not edit!
// Specification: test/f32-rdsum.yaml
// Generator: tools/generate-rdsum-benchmark.py
#include "bench/rsum-benchmark.h"
#include "bench/utils.h"
#include <benchmark/benchmark.h>
#include <xnnpack.h>
#include <xnnpack/aligned-allocator.h>
#include <xnnpack/common.h>
#include <xnnpack/reduce.h>
#include <xnnpack/microfnptr.h>
#include <xnnpack/microparams-init.h>
BENCHMARK_CAPTURE(f32_rsum_discontig, scalar_c4,
xnn_f32_rdsum_ukernel_7p7x__scalar_c4,
xnn_init_f32_scale_scalar_params)
->Apply(BenchmarkBatch)
->UseRealTime();
#if XNN_ARCH_ARM || XNN_ARCH_ARM64
BENCHMARK_CAPTURE(f32_rsum_discontig, neon_c16,
xnn_f32_rdsum_ukernel_7p7x__neon_c16,
xnn_init_f32_scale_scalar_params,
benchmark::utils::CheckNEON)
->Apply(BenchmarkBatch)
->UseRealTime();
#endif // XNN_ARCH_ARM || XNN_ARCH_ARM64
#if XNN_ARCH_ARM || XNN_ARCH_ARM64
BENCHMARK_CAPTURE(f32_rsum_discontig, neon_c32,
xnn_f32_rdsum_ukernel_7p7x__neon_c32,
xnn_init_f32_scale_scalar_params,
benchmark::utils::CheckNEON)
->Apply(BenchmarkBatch)
->UseRealTime();
#endif // XNN_ARCH_ARM || XNN_ARCH_ARM64
#if XNN_ARCH_ARM || XNN_ARCH_ARM64
BENCHMARK_CAPTURE(f32_rsum_discontig, neon_c64,
xnn_f32_rdsum_ukernel_7p7x__neon_c64,
xnn_init_f32_scale_scalar_params,
benchmark::utils::CheckNEON)
->Apply(BenchmarkBatch)
->UseRealTime();
#endif // XNN_ARCH_ARM || XNN_ARCH_ARM64
#if XNN_ARCH_X86 || XNN_ARCH_X86_64
BENCHMARK_CAPTURE(f32_rsum_discontig, sse_c16,
xnn_f32_rdsum_ukernel_7p7x__sse_c16,
xnn_init_f32_scale_sse_params)
->Apply(BenchmarkBatch)
->UseRealTime();
#endif // XNN_ARCH_X86 || XNN_ARCH_X86_64
#if XNN_ARCH_X86 || XNN_ARCH_X86_64
BENCHMARK_CAPTURE(f32_rsum_discontig, sse_c32,
xnn_f32_rdsum_ukernel_7p7x__sse_c32,
xnn_init_f32_scale_sse_params)
->Apply(BenchmarkBatch)
->UseRealTime();
#endif // XNN_ARCH_X86 || XNN_ARCH_X86_64
#if XNN_ARCH_X86 || XNN_ARCH_X86_64
BENCHMARK_CAPTURE(f32_rsum_discontig, sse_c64,
xnn_f32_rdsum_ukernel_7p7x__sse_c64,
xnn_init_f32_scale_sse_params)
->Apply(BenchmarkBatch)
->UseRealTime();
#endif // XNN_ARCH_X86 || XNN_ARCH_X86_64
#if XNN_ARCH_X86 || XNN_ARCH_X86_64
BENCHMARK_CAPTURE(f32_rsum_discontig, avx_c16,
xnn_f32_rdsum_ukernel_7p7x__avx_c16,
xnn_init_f32_scale_avx_params,
benchmark::utils::CheckAVX)
->Apply(BenchmarkBatch)
->UseRealTime();
#endif // XNN_ARCH_X86 || XNN_ARCH_X86_64
#if XNN_ARCH_X86 || XNN_ARCH_X86_64
BENCHMARK_CAPTURE(f32_rsum_discontig, avx_c32,
xnn_f32_rdsum_ukernel_7p7x__avx_c32,
xnn_init_f32_scale_avx_params,
benchmark::utils::CheckAVX)
->Apply(BenchmarkBatch)
->UseRealTime();
#endif // XNN_ARCH_X86 || XNN_ARCH_X86_64
#if XNN_ARCH_X86 || XNN_ARCH_X86_64
BENCHMARK_CAPTURE(f32_rsum_discontig, avx_c64,
xnn_f32_rdsum_ukernel_7p7x__avx_c64,
xnn_init_f32_scale_avx_params,
benchmark::utils::CheckAVX)
->Apply(BenchmarkBatch)
->UseRealTime();
#endif // XNN_ARCH_X86 || XNN_ARCH_X86_64
#ifndef XNNPACK_BENCHMARK_NO_MAIN
BENCHMARK_MAIN();
#endif