vecpow.hpp

Go to the documentation of this file.

#pragma once
 
namespace monolish {
namespace {
template <typename F1, typename F2, typename F3>
void vvpow_core(const F1 &a, const F2 &b, F3 &y) {
  Logger &logger = Logger::get_instance();
  logger.func_in(monolish_func);
 
  // err
  assert(util::is_same_size(a, b, y));
  assert(util::is_same_device_mem_stat(a, b, y));
 
  internal::vpow(y.size(), a.data() + a.get_offset(), b.data() + b.get_offset(),
                 y.data() + y.get_offset(), y.get_device_mem_stat());
 
  logger.func_out();
}
 
template <typename F1, typename F2, typename F3>
void svpow_core(const F1 &a, const F2 &alpha, F3 &y) {
  Logger &logger = Logger::get_instance();
  logger.func_in(monolish_func);
 
  // err
  assert(util::is_same_size(a, y));
  assert(util::is_same_device_mem_stat(a, y));
 
  internal::vpow(y.size(), a.data() + a.get_offset(), alpha,
                 y.data() + y.get_offset(), y.get_device_mem_stat());
 
  logger.func_out();
}
} // namespace
} // namespace monolish