divide_columns.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_OPENCL_KERNELS_DIVIDE_COLUMNS_HPP
#define STAN_MATH_OPENCL_KERNELS_DIVIDE_COLUMNS_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 *divide_columns_kernel_code = STRINGIFY(
    // \endcond
    __kernel void divide_columns_vec(__global double *A, __global double *vec,
                                     int vec_size) {
      const int i = get_global_id(0);
      A[i] /= vec[i % vec_size];
    }
    // \cond
);
// \endcond
const kernel_cl<out_buffer, in_buffer, int> divide_columns_vec(
    "divide_columns_vec", {indexing_helpers, divide_columns_kernel_code});
 
}  // namespace opencl_kernels
}  // namespace math
}  // namespace stan
#endif
#endif