mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-03-21 23:57:39 +00:00
f5ac3ee359
* chore(copyright): update copyright header for tile_engine directory * chore(copyright): update copyright header for script directory * chore(copyright): update copyright header for test_data directory * chore(copyright): update copyright header for python directory * chore(copyright): update copyright header for profiler directory * chore(copyright): update copyright header for library directory * chore(copyright): update copyright header for include directory
44 lines
1.5 KiB
C++
44 lines
1.5 KiB
C++
// Copyright (c) Advanced Micro Devices, Inc., or its affiliates.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
#pragma once
|
|
|
|
#include <algorithm>
|
|
#include <iterator>
|
|
#include <type_traits>
|
|
#include <utility>
|
|
|
|
namespace ck {
|
|
namespace ranges {
|
|
template <typename InputRange, typename OutputIterator>
|
|
auto copy(InputRange&& range,
|
|
OutputIterator iter) -> decltype(std::copy(std::begin(std::forward<InputRange>(range)),
|
|
std::end(std::forward<InputRange>(range)),
|
|
iter))
|
|
{
|
|
return std::copy(std::begin(std::forward<InputRange>(range)),
|
|
std::end(std::forward<InputRange>(range)),
|
|
iter);
|
|
}
|
|
|
|
template <typename T, typename OutputRange>
|
|
auto fill(OutputRange&& range, const T& init)
|
|
-> std::void_t<decltype(std::fill(std::begin(std::forward<OutputRange>(range)),
|
|
std::end(std::forward<OutputRange>(range)),
|
|
init))>
|
|
{
|
|
std::fill(std::begin(std::forward<OutputRange>(range)),
|
|
std::end(std::forward<OutputRange>(range)),
|
|
init);
|
|
}
|
|
|
|
template <typename InputRange, typename OutputIterator, typename UnaryOperation>
|
|
auto transform(InputRange&& range, OutputIterator iter, UnaryOperation unary_op)
|
|
-> decltype(std::transform(std::begin(range), std::end(range), iter, unary_op))
|
|
{
|
|
return std::transform(std::begin(range), std::end(range), iter, unary_op);
|
|
}
|
|
|
|
} // namespace ranges
|
|
} // namespace ck
|