kernel_interface.hh

Go to the documentation of this file.

/*
    @@@@@@@@  @@           @@@@@@   @@@@@@@@ @@
   /@@/////  /@@          @@////@@ @@////// /@@
   /@@       /@@  @@@@@  @@    // /@@       /@@
   /@@@@@@@  /@@ @@///@@/@@       /@@@@@@@@@/@@
   /@@////   /@@/@@@@@@@/@@       ////////@@/@@
   /@@       /@@/@@//// //@@    @@       /@@/@@
   /@@       @@@//@@@@@@ //@@@@@@  @@@@@@@@ /@@
   //       ///  //////   //////  ////////  //

   Copyright (c) 2016, Triad National Security, LLC
   All rights reserved.
                                                                              */
#pragma once

#if !defined(__FLECSI_PRIVATE__)
#error Do not include this file directly!
#endif

#if !defined(FLECSI_ENABLE_KOKKOS)
#error FLECSI_ENABLE_KOKKOS not defined! This file depends on Kokkos!
#endif

#include <kokkos.h>

#include <string>

namespace flecsi {
namespace exec {

template<typename ITERATOR, typename LAMBDA>
parallel_for(ITERATOR const iterator,
  LAMBDA const lambda,
  std::string const & name = "") {

  struct functor_t {

    functor_t(ITERATOR & iterator, LAMBDA & lambda)
      : iterator_(iterator), lambda_(lambda) {}

    KOKKOS_INLINE_FUNCTION void operator()(int i) const {
      lambda(iterator[i]);
    } // operator()

  private:
    ITERATOR & iterator_;
    LAMBDA & lambda_;

  }; // struct functor_t

  Kokkos::parallel_for(name, iterator.size(), functor_t{iterator, lambda});

} // parallel_for

// We will need this when/if we begin to customize behavior based on
// ITERATOR type.
// template<typename ITERATOR> struct forall {};

template<typename ITERATOR>
struct forall {

  forall(ITERATOR iterator, std::string const & name = "")
    : iterator_(iterator), name_(name) {}

  template<typename LAMBDA>
  struct functor {

    functor(ITERATOR iterator, LAMBDA lambda, std::string const & name)
      : iterator_(iterator), lambda_(lambda), name_(name) {}

    void operator()(int64_t i) const {
      lambda_(iterator_[i]);
    } // operator()

  private:
    ITERATOR iterator_;
    LAMBDA lambda_;
    std::string & const name_;

  }; // struct functor_t

  template<typename CALLABLE>
  void operator<<(CALLABLE l) {
    Kokkos::parallel_for(
      name_, iterator_.size(), functor_t{iterator_, lambda_});
  } // operator<<

private:
  ITERATOR iterator_;
  std::string & const name_;

}; // struct forall

} // namespace exec
} // namespace flecsi