wehub-resource-sync
75 lines
5.0 KiB
C++
75 lines
5.0 KiB
C++
/* ******************************************************************************
|
|
*
|
|
*
|
|
* This program and the accompanying materials are made available under the
|
|
* terms of the Apache License, Version 2.0 which is available at
|
|
* https://www.apache.org/licenses/LICENSE-2.0.
|
|
*
|
|
* See the NOTICE file distributed with this work for additional
|
|
* information regarding copyright ownership.
|
|
* 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.
|
|
*
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
******************************************************************************/
|
|
|
|
#pragma once
|
|
#ifndef OP_MACROS_INDEX_REDUCE_H_
|
|
#define OP_MACROS_INDEX_REDUCE_H_
|
|
|
|
#include "op_types.h"
|
|
|
|
namespace simdOps {
|
|
|
|
// =============================================================================
|
|
// INDEX REDUCE OPERATION MACROS
|
|
// =============================================================================
|
|
|
|
/**
|
|
* @brief Declares an index reduce operation with proper SIMD handling
|
|
*/
|
|
#define DECLARE_INDEX_REDUCE_OP(OP_NAME, STARTING_VAL, UPDATE_CONDITION, MERGE_CONDITION) \
|
|
template <typename X, typename Z, \
|
|
typename std::enable_if< \
|
|
std::is_arithmetic<X>::value && \
|
|
std::is_arithmetic<Z>::value, int>::type = 0> \
|
|
class OP_NAME { \
|
|
public: \
|
|
static SD_HOST_DEVICE inline functions::indexreduce::IndexValue<X> op(functions::indexreduce::IndexValue<X> val, \
|
|
X* extraParams) { \
|
|
return val; \
|
|
} \
|
|
static SD_HOST_DEVICE inline X startingValue(const X* input) { return STARTING_VAL; } \
|
|
static SD_HOST_DEVICE inline functions::indexreduce::IndexValue<X> startingIndexValue(const X* input) { \
|
|
functions::indexreduce::IndexValue<X> local; \
|
|
local.value = startingValue(input); \
|
|
local.index = 0; \
|
|
return local; \
|
|
} \
|
|
static SD_HOST_DEVICE inline functions::indexreduce::IndexValue<X> update( \
|
|
functions::indexreduce::IndexValue<X>& old, functions::indexreduce::IndexValue<X>& opOutput, \
|
|
X* extraParams) { \
|
|
if (UPDATE_CONDITION) return opOutput; \
|
|
return old; \
|
|
} \
|
|
static SD_HOST_DEVICE inline functions::indexreduce::IndexValue<X> merge( \
|
|
functions::indexreduce::IndexValue<X> f1, functions::indexreduce::IndexValue<X> f2, X* extraParams) { \
|
|
if (MERGE_CONDITION) return f2; \
|
|
return f1; \
|
|
} \
|
|
static SD_HOST_DEVICE inline functions::indexreduce::IndexValue<X> postProcess( \
|
|
functions::indexreduce::IndexValue<X> reduction, int n, int xOffset, X* dx, int incx, X* extraParams, \
|
|
X* result) { \
|
|
return reduction; \
|
|
} \
|
|
static SD_HOST_DEVICE inline functions::indexreduce::IndexValue<X> op( \
|
|
functions::indexreduce::IndexValue<X> d1, functions::indexreduce::IndexValue<X> d2, X* extraParams) { \
|
|
return d1; \
|
|
} \
|
|
};
|
|
} // namespace simdOps
|
|
|
|
#endif // OP_MACROS_INDEX_REDUCE_H_
|