stdgpu/doxygen/numeric_8h_source.html

/*

 *  Copyright 2022 Patrick Stotko

 *  Licensed under the Apache License, Version 2.0 (the "License");

 *  you may not use this file except in compliance with the License.

 *  You may obtain a copy of the License at

 *

 *      http://www.apache.org/licenses/LICENSE-2.0

 *

 *  Unless required by applicable law or agreed to in writing, software

 *  distributed under the License is distributed on an "AS IS" BASIS,

 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 *  See the License for the specific language governing permissions and

 *  limitations under the License.

 */


#ifndef STDGPU_NUMERIC_H

#define STDGPU_NUMERIC_H


#include <stdgpu/execution.h>


namespace stdgpu

{


template <typename ExecutionPolicy,

          typename Iterator,

          typename T,

          STDGPU_DETAIL_OVERLOAD_IF(is_execution_policy_v<remove_cvref_t<ExecutionPolicy>>)>

void

iota(ExecutionPolicy&& policy, Iterator begin, Iterator end, T value);


template <typename IndexType,

          typename ExecutionPolicy,

          typename T,

          typename BinaryFunction,

          typename UnaryFunction,

          STDGPU_DETAIL_OVERLOAD_IF(is_execution_policy_v<remove_cvref_t<ExecutionPolicy>>)>

T

transform_reduce_index(ExecutionPolicy&& policy, IndexType size, T init, BinaryFunction reduce, UnaryFunction f);


} // namespace stdgpu


#include <stdgpu/impl/numeric_detail.h>


#endif // STDGPU_NUMERIC_H

execution.h

stdgpu::size
index64_t size(T *array)
Finds the size (number of elements) of the given dynamically allocated array.

stdgpu::transform_reduce_index
T transform_reduce_index(ExecutionPolicy &&policy, IndexType size, T init, BinaryFunction reduce, UnaryFunction f)
Calls the given unary function with an index from the range [0, size) and performs a reduction afterw...

stdgpu::iota
void iota(ExecutionPolicy &&policy, Iterator begin, Iterator end, T value)
Writes ascending values {values + i} to the i-th position of the given range.