mirror of
https://github.com/matrix-construct/construct
synced 2024-11-26 08:42:34 +01:00
73 lines
1.9 KiB
C++
73 lines
1.9 KiB
C++
// The Construct
|
|
//
|
|
// Copyright (C) The Construct Developers, Authors & Contributors
|
|
// Copyright (C) 2016-2020 Jason Volk <jason@zemos.net>
|
|
//
|
|
// Permission to use, copy, modify, and/or distribute this software for any
|
|
// purpose with or without fee is hereby granted, provided that the above
|
|
// copyright notice and this permission notice is present in all copies. The
|
|
// full license for this software is available in the LICENSE file.
|
|
|
|
#pragma once
|
|
#define HAVE_IRCD_SIMD_ACCUMULATE_H
|
|
|
|
namespace ircd::simd
|
|
{
|
|
/// Accumulate into block_t by synthetic-reference. The closure has an
|
|
/// opportunity to update the accumulator while data blocks are being
|
|
/// streamed from the input. The mask indicates which elements of the
|
|
/// input block are valid if the input is smaller or offset from the bs.
|
|
template<class block_t>
|
|
using accumulate_prototype = void (block_t &, block_t, block_t mask);
|
|
|
|
template<class block_t,
|
|
class input_t,
|
|
class lambda>
|
|
block_t accumulate(const input_t *, const u64x2, block_t, lambda&&) noexcept;
|
|
|
|
template<class block_t,
|
|
class lambda>
|
|
block_t accumulate(const const_buffer &, block_t, lambda&&) noexcept;
|
|
}
|
|
|
|
/// Streaming accumulation
|
|
///
|
|
template<class block_t,
|
|
class lambda>
|
|
inline block_t
|
|
ircd::simd::accumulate(const const_buffer &buf,
|
|
block_t val,
|
|
lambda&& closure)
|
|
noexcept
|
|
{
|
|
const u64x2 max
|
|
{
|
|
0, size(buf),
|
|
};
|
|
|
|
return accumulate(data(buf), max, val, std::forward<lambda>(closure));
|
|
}
|
|
|
|
/// Streaming accumulation
|
|
///
|
|
template<class block_t,
|
|
class input_t,
|
|
class lambda>
|
|
inline block_t
|
|
ircd::simd::accumulate(const input_t *const __restrict__ in,
|
|
const u64x2 max,
|
|
block_t val,
|
|
lambda&& closure)
|
|
noexcept
|
|
{
|
|
const u64x2 res
|
|
{
|
|
for_each<block_t>(in, max, [&val, &closure]
|
|
(const auto block, const auto mask)
|
|
{
|
|
closure(val, block, mask);
|
|
})
|
|
};
|
|
|
|
return val;
|
|
}
|