blob: 8b5803d10517cf93bcb441e13f3dfbe1c5ecc383 [file] [log] [blame]
Nicolas Capensf7e7cc52022-02-06 08:51:05 -05001// Copyright 2022 The SwiftShader Authors. All Rights Reserved.
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7// http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
Nicolas Capens93af45d2022-02-06 08:52:56 -050015#include "ShaderCore.hpp"
16#include "Reactor/Reactor.hpp"
Nicolas Capensf7e7cc52022-02-06 08:51:05 -050017
18#include "benchmark/benchmark.h"
19
Nicolas Capensece389a2022-02-06 09:49:19 -050020#include <vector>
21
Nicolas Capensf7e7cc52022-02-06 08:51:05 -050022using namespace rr;
23
24BENCHMARK_MAIN();
25
26// Macro that creates a lambda wrapper around the input overloaded function,
27// creating a non-overload based on the args. This is useful for passing
28// overloaded functions as template arguments.
29// See https://stackoverflow.com/questions/25871381/c-overloaded-function-as-template-argument
30#define LIFT(fname) \
31 [](auto &&...args) -> decltype(auto) { \
32 return fname(std::forward<decltype(args)>(args)...); \
33 }
34
35template<typename Func, class... Args>
Nicolas Capensece389a2022-02-06 09:49:19 -050036static void Transcendental1(benchmark::State &state, Func func, Args &&...args)
Nicolas Capensf7e7cc52022-02-06 08:51:05 -050037{
Nicolas Capensece389a2022-02-06 09:49:19 -050038 const int REPS = state.range(0);
39
40 FunctionT<void(float *, float *)> function;
Nicolas Capensf7e7cc52022-02-06 08:51:05 -050041 {
Nicolas Capensece389a2022-02-06 09:49:19 -050042 Pointer<Float4> r = Pointer<Float>(function.Arg<0>());
43 Pointer<Float4> a = Pointer<Float>(function.Arg<1>());
44
45 for(int i = 0; i < REPS; i++)
46 {
47 r[i] = func(a[i], args...);
48 }
Nicolas Capensf7e7cc52022-02-06 08:51:05 -050049 }
50
51 auto routine = function("one");
52
Nicolas Capensece389a2022-02-06 09:49:19 -050053 std::vector<float> r(REPS * 4);
54 std::vector<float> a(REPS * 4, 1.0f);
55
Nicolas Capensf7e7cc52022-02-06 08:51:05 -050056 for(auto _ : state)
57 {
Nicolas Capensece389a2022-02-06 09:49:19 -050058 routine(r.data(), a.data());
Nicolas Capensf7e7cc52022-02-06 08:51:05 -050059 }
60}
61
62template<typename Func, class... Args>
Nicolas Capensece389a2022-02-06 09:49:19 -050063static void Transcendental2(benchmark::State &state, Func func, Args &&...args)
Nicolas Capensf7e7cc52022-02-06 08:51:05 -050064{
Nicolas Capensece389a2022-02-06 09:49:19 -050065 const int REPS = state.range(0);
66
67 FunctionT<void(float *, float *, float *)> function;
Nicolas Capensf7e7cc52022-02-06 08:51:05 -050068 {
Nicolas Capensece389a2022-02-06 09:49:19 -050069 Pointer<Float4> r = Pointer<Float>(function.Arg<0>());
70 Pointer<Float4> a = Pointer<Float>(function.Arg<1>());
71 Pointer<Float4> b = Pointer<Float>(function.Arg<2>());
72
73 for(int i = 0; i < REPS; i++)
74 {
75 r[i] = func(a[i], b[i], args...);
76 }
Nicolas Capensf7e7cc52022-02-06 08:51:05 -050077 }
78
79 auto routine = function("two");
80
Nicolas Capensece389a2022-02-06 09:49:19 -050081 std::vector<float> r(REPS * 4);
82 std::vector<float> a(REPS * 4, 0.456f);
83 std::vector<float> b(REPS * 4, 0.789f);
84
Nicolas Capensf7e7cc52022-02-06 08:51:05 -050085 for(auto _ : state)
86 {
Nicolas Capensece389a2022-02-06 09:49:19 -050087 routine(r.data(), a.data(), b.data());
Nicolas Capensf7e7cc52022-02-06 08:51:05 -050088 }
89}
90
Nicolas Capensece389a2022-02-06 09:49:19 -050091// No operation; just copy the input to the output, for use as a baseline.
92static Float4 Nop(RValue<Float4> x)
93{
94 return x;
95}
Nicolas Capensf7e7cc52022-02-06 08:51:05 -050096
Nicolas Capensece389a2022-02-06 09:49:19 -050097static const int REPS = 10;
Nicolas Capensf7e7cc52022-02-06 08:51:05 -050098
Nicolas Capensece389a2022-02-06 09:49:19 -050099BENCHMARK_CAPTURE(Transcendental1, Nop, Nop)->Arg(REPS);
Nicolas Capensf7e7cc52022-02-06 08:51:05 -0500100
Nicolas Capensece389a2022-02-06 09:49:19 -0500101BENCHMARK_CAPTURE(Transcendental1, rr_Sin, rr::Sin)->Arg(REPS);
102BENCHMARK_CAPTURE(Transcendental1, sw_Sin, sw::Sin)->Arg(REPS);
103BENCHMARK_CAPTURE(Transcendental1, rr_Cos, rr::Cos)->Arg(REPS);
104BENCHMARK_CAPTURE(Transcendental1, sw_Cos, sw::Cos)->Arg(REPS);
105BENCHMARK_CAPTURE(Transcendental1, rr_Tan, rr::Tan)->Arg(REPS);
106BENCHMARK_CAPTURE(Transcendental1, sw_Tan, sw::Tan)->Arg(REPS);
Nicolas Capensf7e7cc52022-02-06 08:51:05 -0500107
Nicolas Capensece389a2022-02-06 09:49:19 -0500108BENCHMARK_CAPTURE(Transcendental1, rr_Asin, rr::Asin)->Arg(REPS);
109BENCHMARK_CAPTURE(Transcendental1, sw_Asin_highpp, sw::Asin, false /* relaxedPrecision */)->Arg(REPS);
110BENCHMARK_CAPTURE(Transcendental1, sw_Asin_relaxedp, sw::Asin, true /* relaxedPrecision */)->Arg(REPS);
111BENCHMARK_CAPTURE(Transcendental1, rr_Acos, rr::Acos)->Arg(REPS);
112BENCHMARK_CAPTURE(Transcendental1, sw_Acos_highp, sw::Acos, false /* relaxedPrecision */)->Arg(REPS);
113BENCHMARK_CAPTURE(Transcendental1, sw_Acos_relaxedp, sw::Acos, true /* relaxedPrecision */)->Arg(REPS);
Nicolas Capensf7e7cc52022-02-06 08:51:05 -0500114
Nicolas Capensece389a2022-02-06 09:49:19 -0500115BENCHMARK_CAPTURE(Transcendental1, rr_Atan, rr::Atan)->Arg(REPS);
116BENCHMARK_CAPTURE(Transcendental1, sw_Atan, sw::Atan)->Arg(REPS);
117BENCHMARK_CAPTURE(Transcendental1, rr_Sinh, rr::Sinh)->Arg(REPS);
118BENCHMARK_CAPTURE(Transcendental1, sw_Sinh, sw::Sinh)->Arg(REPS);
119BENCHMARK_CAPTURE(Transcendental1, rr_Cosh, rr::Cosh)->Arg(REPS);
120BENCHMARK_CAPTURE(Transcendental1, sw_Cosh, sw::Cosh)->Arg(REPS);
121BENCHMARK_CAPTURE(Transcendental1, rr_Tanh, rr::Tanh)->Arg(REPS);
122BENCHMARK_CAPTURE(Transcendental1, sw_Tanh, sw::Tanh)->Arg(REPS);
123
124BENCHMARK_CAPTURE(Transcendental1, rr_Asinh, rr::Asinh)->Arg(REPS);
125BENCHMARK_CAPTURE(Transcendental1, sw_Asinh, sw::Asinh)->Arg(REPS);
126BENCHMARK_CAPTURE(Transcendental1, rr_Acosh, rr::Acosh)->Arg(REPS);
127BENCHMARK_CAPTURE(Transcendental1, sw_Acosh, sw::Acosh)->Arg(REPS);
128BENCHMARK_CAPTURE(Transcendental1, rr_Atanh, rr::Atanh)->Arg(REPS);
129BENCHMARK_CAPTURE(Transcendental1, sw_Atanh, sw::Atanh)->Arg(REPS);
130BENCHMARK_CAPTURE(Transcendental2, rr_Atan2, rr::Atan2)->Arg(REPS);
131BENCHMARK_CAPTURE(Transcendental2, sw_Atan2, sw::Atan2)->Arg(REPS);
132
133BENCHMARK_CAPTURE(Transcendental2, rr_Pow, rr::Pow)->Arg(REPS);
134BENCHMARK_CAPTURE(Transcendental2, sw_Pow, sw::Pow)->Arg(REPS);
135BENCHMARK_CAPTURE(Transcendental1, rr_Exp, rr::Exp)->Arg(REPS);
136BENCHMARK_CAPTURE(Transcendental1, sw_Exp, sw::Exp)->Arg(REPS);
137BENCHMARK_CAPTURE(Transcendental1, rr_Log, rr::Log)->Arg(REPS);
138BENCHMARK_CAPTURE(Transcendental1, sw_Log, sw::Log)->Arg(REPS);
139BENCHMARK_CAPTURE(Transcendental1, rr_Exp2, LIFT(rr::Exp2))->Arg(REPS);
140BENCHMARK_CAPTURE(Transcendental1, sw_Exp2, LIFT(sw::Exp2))->Arg(REPS);
141BENCHMARK_CAPTURE(Transcendental1, rr_Log2, LIFT(rr::Log2))->Arg(REPS);
142BENCHMARK_CAPTURE(Transcendental1, sw_Log2, LIFT(sw::Log2))->Arg(REPS);
143
144BENCHMARK_CAPTURE(Transcendental1, rr_Rcp_pp_exactAtPow2_true, LIFT(Rcp_pp), true)->Arg(REPS);
145BENCHMARK_CAPTURE(Transcendental1, rr_Rcp_pp_exactAtPow2_false, LIFT(Rcp_pp), false)->Arg(REPS);
146BENCHMARK_CAPTURE(Transcendental1, rr_RcpSqrt_pp, LIFT(RcpSqrt_pp))->Arg(REPS);