transpose.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_OPENCL_KERNEL_GENERATOR_TRANSPOSE_HPP
#define STAN_MATH_OPENCL_KERNEL_GENERATOR_TRANSPOSE_HPP
#ifdef STAN_OPENCL
 
#include <stan/math/opencl/matrix_cl_view.hpp>
#include <stan/math/prim/meta.hpp>
#include <stan/math/opencl/kernel_generator/type_str.hpp>
#include <stan/math/opencl/kernel_generator/name_generator.hpp>
#include <stan/math/opencl/kernel_generator/operation_cl_lhs.hpp>
#include <stan/math/opencl/kernel_generator/as_operation_cl.hpp>
#include <algorithm>
#include <string>
#include <set>
#include <tuple>
#include <type_traits>
#include <utility>
 
namespace stan {
namespace math {
 
template <typename Arg>
class transpose_
    : public operation_cl_lhs<
          transpose_<Arg>, typename std::remove_reference_t<Arg>::Scalar, Arg> {
 public:
  using Scalar = typename std::remove_reference_t<Arg>::Scalar;
  using base = operation_cl_lhs<transpose_<Arg>, Scalar, Arg>;
  using base::var_name_;
  using view_transitivity = std::tuple<std::true_type>;
  using base::operator=;
 
  explicit transpose_(Arg&& a) : base(std::forward<Arg>(a)) {}
 
  inline auto deep_copy() const {
    auto&& arg_copy = this->template get_arg<0>().deep_copy();
    return transpose_<std::remove_reference_t<decltype(arg_copy)>>{
        std::move(arg_copy)};
  }
 
  inline void modify_argument_indices(std::string& row_index_name,
                                      std::string& col_index_name) const {
    std::swap(row_index_name, col_index_name);
  }
 
  inline int rows() const { return this->template get_arg<0>().cols(); }
 
  inline int cols() const { return this->template get_arg<0>().rows(); }
 
  inline std::pair<int, int> extreme_diagonals() const {
    std::pair<int, int> arg_diags
        = this->template get_arg<0>().extreme_diagonals();
    return {-arg_diags.second, -arg_diags.first};
  }
 
  void set_view(int bottom_diagonal, int top_diagonal, int bottom_zero_diagonal,
                int top_zero_diagonal) const {
    this->template get_arg<0>().set_view(
        top_diagonal, bottom_diagonal, top_zero_diagonal, bottom_zero_diagonal);
  }
 
  void check_assign_dimensions(int rows, int cols) const {
    this->template get_arg<0>().check_assign_dimensions(cols, rows);
  }
};
 
template <typename Arg,
          typename = require_all_kernel_expressions_and_none_scalar_t<Arg>>
inline auto transpose(Arg&& a) {
  auto&& a_operation = as_operation_cl(std::forward<Arg>(a)).deep_copy();
  return transpose_<std::remove_reference_t<decltype(a_operation)>>{
      std::move(a_operation)};
}
}  // namespace math
}  // namespace stan
 
#endif
#endif