check_symmetric.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_OPENCL_KERNELS_CHECK_SYMMETRIC_HPP
#define STAN_MATH_OPENCL_KERNELS_CHECK_SYMMETRIC_HPP
#ifdef STAN_OPENCL
 
#include <stan/math/opencl/kernel_cl.hpp>
#include <stan/math/opencl/buffer_types.hpp>
#include <string>
 
namespace stan {
namespace math {
namespace opencl_kernels {
// \cond
static constexpr const char *is_symmetric_kernel_code = STRINGIFY(
    // \endcond
    __kernel void is_symmetric(__global double *A, __global int *flag,
                               unsigned int rows, unsigned int cols,
                               double tolerance) {
      const int i = get_global_id(0);
      const int j = get_global_id(1);
      if (i < rows && j < cols) {
        double diff = fabs(A(i, j) - A(j, i));
        if (diff > tolerance) {
          flag[0] = 0;
        }
      }
    }
    // \cond
);
// \endcond
 
const kernel_cl<in_buffer, out_buffer, int, int, const double> check_symmetric(
    "is_symmetric", {indexing_helpers, is_symmetric_kernel_code});
 
}  // namespace opencl_kernels
}  // namespace math
}  // namespace stan
#endif
#endif