// ================================================================================================= // This file is part of the CLBlast project. The project is licensed under Apache Version 2.0. This // project loosely follows the Google C++ styleguide and uses a tab-size of two spaces and a max- // width of 100 characters per line. // // Author(s): // Cedric Nugteren // // This file implements the Xher class (see the header for information about the class). // // ================================================================================================= #include "routines/level2/xher.hpp" #include namespace clblast { // ================================================================================================= // Constructor: forwards to base class constructor template Xher::Xher(Queue &queue, EventPointer event, const std::string &name): Routine(queue, event, name, {"Xger"}, PrecisionValue(), {}, { #include "../../kernels/level2/level2.opencl" #include "../../kernels/level2/xher.opencl" }) { } // ================================================================================================= // Specializations to compute alpha of type 'T' template <> float2 Xher::GetAlpha(const float alpha) { return float2{alpha, 0.0f}; } template <> double2 Xher::GetAlpha(const double alpha) { return double2{alpha, 0.0}; } template <> float Xher::GetAlpha(const float alpha) { return alpha; } template <> double Xher::GetAlpha(const double alpha) { return alpha; } template <> half Xher::GetAlpha(const half alpha) { return alpha; } // ================================================================================================= // The main routine template void Xher::DoHer(const Layout layout, const Triangle triangle, const size_t n, const U alpha, const Buffer &x_buffer, const size_t x_offset, const size_t x_inc, const Buffer &a_buffer, const size_t a_offset, const size_t a_ld, const bool packed) { // Makes sure the dimensions are larger than zero if (n == 0) { throw BLASError(StatusCode::kInvalidDimension); } // The data is either in the upper or lower triangle const auto is_upper = ((triangle == Triangle::kUpper && layout != Layout::kRowMajor) || (triangle == Triangle::kLower && layout == Layout::kRowMajor)); const auto is_rowmajor = (layout == Layout::kRowMajor); // Tests the matrix and the vectors for validity if (packed) { TestMatrixAP(n, a_buffer, a_offset); } else { TestMatrixA(n, n, a_buffer, a_offset, a_ld); } TestVectorX(n, x_buffer, x_offset, x_inc); // If alpha is zero an update is not required if (alpha == U{0}) { return; } // Creates a matching version of alpha const auto matching_alpha = GetAlpha(alpha); // Retrieves the kernel from the compiled binary auto kernel = Kernel(program_, "Xher"); // Sets the kernel arguments kernel.SetArgument(0, static_cast(n)); kernel.SetArgument(1, GetRealArg(matching_alpha)); kernel.SetArgument(2, x_buffer()); kernel.SetArgument(3, static_cast(x_offset)); kernel.SetArgument(4, static_cast(x_inc)); kernel.SetArgument(5, a_buffer()); kernel.SetArgument(6, static_cast(a_offset)); kernel.SetArgument(7, static_cast(a_ld)); kernel.SetArgument(8, static_cast(is_upper)); kernel.SetArgument(9, static_cast(is_rowmajor)); // Launches the kernel auto global_one = Ceil(CeilDiv(n, db_["WPT"]), db_["WGS1"]); auto global_two = Ceil(CeilDiv(n, db_["WPT"]), db_["WGS2"]); auto global = std::vector{global_one, global_two}; auto local = std::vector{db_["WGS1"], db_["WGS2"]}; RunKernel(kernel, queue_, device_, global, local, event_); } // ================================================================================================= // Compiles the templated class template class Xher; template class Xher; template class Xher; template class Xher; template class Xher; // ================================================================================================= } // namespace clblast