From dbdb58c6002cbd693f246f1e93919cc32ad4055a Mon Sep 17 00:00:00 2001 From: CNugteren Date: Sun, 9 Aug 2015 15:50:41 +0200 Subject: Refactored the tuners, added JSON output --- src/tuning/transpose.cc | 139 ++++++++++++++++++++++++++++++------------------ 1 file changed, 86 insertions(+), 53 deletions(-) (limited to 'src/tuning/transpose.cc') diff --git a/src/tuning/transpose.cc b/src/tuning/transpose.cc index 8963a688..2ffdb7aa 100644 --- a/src/tuning/transpose.cc +++ b/src/tuning/transpose.cc @@ -7,13 +7,12 @@ // Author(s): // Cedric Nugteren // -// This file implements an auto-tuner to tune the transpose OpenCL kernels. It uses CLTune. +// This file uses the CLTune auto-tuner to tune the transpose OpenCL kernels. // // ================================================================================================= #include #include -#include #include "internal/utilities.h" #include "internal/tuning.h" @@ -21,67 +20,101 @@ namespace clblast { // ================================================================================================= -// The transpose auto-tuner +// See comment at top of file for a description of the class template -void TransposeTune(const Arguments &args, - const std::vector &a_mat, std::vector &b_mat, - cltune::Tuner &tuner) { - - // This points to the PadTransposeMatrix kernel as found in the CLBlast library. This is just one - // example of a transpose kernel. However, all kernels use the same tuning parameters, so one has - // to be chosen as a representative. - std::string sources = - #include "../src/kernels/common.opencl" - #include "../src/kernels/transpose.opencl" - ; - auto id = tuner.AddKernelFromString(sources, "TransposeMatrix", {args.m, args.n}, {1, 1}); - tuner.SetReferenceFromString(sources, "TransposeMatrix", {args.m, args.n}, {8, 8}); - - // Sets the tunable parameters and their possible values - tuner.AddParameter(id, "TRA_DIM", {4, 8, 16, 32, 64}); - tuner.AddParameter(id, "TRA_WPT", {1, 2, 4, 8, 16}); - tuner.AddParameter(id, "TRA_PAD", {0, 1}); - tuner.AddParameter(id, "TRA_SHUFFLE", {0, 1}); - - // Tests for a specific precision - tuner.AddParameter(id, "PRECISION", {static_cast(args.precision)}); - tuner.AddParameterReference("PRECISION", static_cast(args.precision)); - - // Sets the constraints for local memory size limitations - auto LocalMemorySize = [args] (std::vector v) { - return ((v[0]*v[1]*(v[0]*v[1]+v[2]))*GetBytes(args.precision)); - }; - tuner.SetLocalMemoryUsage(id, LocalMemorySize, {"TRA_DIM", "TRA_WPT", "TRA_PAD"}); - - // Modifies the thread-sizes (both global and local) based on the parameters - tuner.DivGlobalSize(id, {"TRA_WPT", "TRA_WPT"}); - tuner.MulLocalSize(id, {"TRA_DIM", "TRA_DIM"}); - - // Sets the function's arguments - tuner.AddArgumentScalar(static_cast(args.m)); - tuner.AddArgumentInput(a_mat); - tuner.AddArgumentOutput(b_mat); -} +class TuneTranspose { + public: + + // The representative kernel and the source code + static std::string KernelFamily() { return "transpose"; } + static std::string KernelName() { return "TransposeMatrix"; } + static std::string GetSources() { + return + #include "../src/kernels/common.opencl" + #include "../src/kernels/transpose.opencl" + ; + } -// ================================================================================================= + // The list of arguments relevant for this routine + static std::vector GetOptions() { return {kArgM, kArgN}; } + + // Tests for valid arguments + static void TestValidArguments(const Arguments &) { } + + // Sets the default values for the arguments + static size_t DefaultM() { return 1024; } + static size_t DefaultN() { return 1024; } + static size_t DefaultK() { return 1; } // N/A for this kernel + static double DefaultFraction() { return 1.0; } // N/A for this kernel + + // Describes how to obtain the sizes of the buffers + static size_t GetSizeX(const Arguments &) { return 1; } // N/A for this kernel + static size_t GetSizeY(const Arguments &) { return 1; } // N/A for this kernel + static size_t GetSizeA(const Arguments &args) { return args.m * args.n; } + static size_t GetSizeB(const Arguments &args) { return args.m * args.n; } + static size_t GetSizeC(const Arguments &) { return 1; } // N/A for this kernel + + // Sets the tuning parameters and their possible values + static void SetParameters(cltune::Tuner &tuner, const size_t id) { + tuner.AddParameter(id, "TRA_DIM", {4, 8, 16, 32, 64}); + tuner.AddParameter(id, "TRA_WPT", {1, 2, 4, 8, 16}); + tuner.AddParameter(id, "TRA_PAD", {0, 1}); + tuner.AddParameter(id, "TRA_SHUFFLE", {0, 1}); + } -// Main function which calls the common client code with the routine-specific function as argument. -void TunerTranspose(int argc, char *argv[]) { - switch(GetPrecision(argc, argv)) { - case Precision::kHalf: throw std::runtime_error("Unsupported precision mode"); - case Precision::kSingle: TunerAB(argc, argv, TransposeTune); break; - case Precision::kDouble: TunerAB(argc, argv, TransposeTune); break; - case Precision::kComplexSingle: TunerAB(argc, argv, TransposeTune); break; - case Precision::kComplexDouble: TunerAB(argc, argv, TransposeTune); break; + // Sets the constraints and local memory size + static void SetConstraints(cltune::Tuner &, const size_t) { } + static void SetLocalMemorySize(cltune::Tuner &tuner, const size_t id, const Arguments &args) { + auto LocalMemorySize = [args] (std::vector v) { + return ((v[0]*v[1]*(v[0]*v[1]+v[2]))*GetBytes(args.precision)); + }; + tuner.SetLocalMemoryUsage(id, LocalMemorySize, {"TRA_DIM", "TRA_WPT", "TRA_PAD"}); } -} + + // Sets the base thread configuration + static std::vector GlobalSize(const Arguments &args) { return {args.m, args.n}; } + static std::vector LocalSize() { return {1, 1}; } + static std::vector LocalSizeRef() { return {8, 8}; } + + // Transforms the thread configuration based on the parameters + using TransformVector = std::vector>; + static TransformVector MulLocal() { return {{"TRA_DIM", "TRA_DIM"}}; } + static TransformVector DivLocal() { return {}; } + static TransformVector MulGlobal() { return {}; } + static TransformVector DivGlobal() { return {{"TRA_WPT", "TRA_WPT"}}; } + + // Sets the kernel's arguments + static void SetArguments(cltune::Tuner &tuner, const Arguments &args, + std::vector &, std::vector &, + std::vector &a_mat, std::vector &b_mat, std::vector &) { + tuner.AddArgumentScalar(static_cast(args.m)); + tuner.AddArgumentInput(a_mat); + tuner.AddArgumentOutput(b_mat); + } + + // Describes how to compute the performance metrics + static size_t GetMetric(const Arguments &args) { + return 2 * args.m * args.n * GetBytes(args.precision); + } + static std::string PerformanceUnit() { return "GB/s"; } +}; // ================================================================================================= } // namespace clblast +// Shortcuts to the clblast namespace +using float2 = clblast::float2; +using double2 = clblast::double2; + // Main function (not within the clblast namespace) int main(int argc, char *argv[]) { - clblast::TunerTranspose(argc, argv); + switch(clblast::GetPrecision(argc, argv)) { + case clblast::Precision::kHalf: throw std::runtime_error("Unsupported precision mode"); + case clblast::Precision::kSingle: clblast::Tuner, float>(argc, argv); break; + case clblast::Precision::kDouble: clblast::Tuner, double>(argc, argv); break; + case clblast::Precision::kComplexSingle: clblast::Tuner, float2>(argc, argv); break; + case clblast::Precision::kComplexDouble: clblast::Tuner, double2>(argc, argv); break; + } return 0; } -- cgit v1.2.3