fill.hpp

// SPDX-FileCopyrightText: Intel Corporation
//
// SPDX-License-Identifier: BSD-3-Clause
 
#pragma once
 
#include <algorithm>
#include <execution>
#include <type_traits>
#include <utility>
 
#include <dr/concepts/concepts.hpp>
#include <dr/detail/logger.hpp>
#include <dr/detail/onedpl_direct_iterator.hpp>
#include <dr/detail/ranges_shim.hpp>
#include <dr/mp/global.hpp>
 
namespace dr::mp {
 
auto fill(dr::distributed_contiguous_range auto &&dr, auto value) {
  for_each(dr, [=](auto &v) { v = value; });
  return rng::end(dr);
}
 
template <dr::distributed_iterator DI>
auto fill(DI first, DI last, auto value) {
  mp::fill(rng::subrange(first, last), value);
  return last;
}
 
} // namespace dr::mp