mirror of
https://gitlab.kitware.com/vtk/vtk-m
synced 2024-10-08 11:29:02 +00:00
305 lines
11 KiB
C++
305 lines
11 KiB
C++
//============================================================================
|
|
// Copyright (c) Kitware, Inc.
|
|
// All rights reserved.
|
|
// See LICENSE.txt for details.
|
|
//
|
|
// This software is distributed WITHOUT ANY WARRANTY; without even
|
|
// the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
|
|
// PURPOSE. See the above copyright notice for more information.
|
|
//============================================================================
|
|
#ifndef vtk_m_exec_AtomicArrayExecutionObject_h
|
|
#define vtk_m_exec_AtomicArrayExecutionObject_h
|
|
|
|
#include <vtkm/Atomic.h>
|
|
#include <vtkm/List.h>
|
|
#include <vtkm/cont/ArrayHandle.h>
|
|
#include <vtkm/cont/DeviceAdapter.h>
|
|
|
|
#include <type_traits>
|
|
|
|
namespace vtkm
|
|
{
|
|
namespace exec
|
|
{
|
|
|
|
namespace detail
|
|
{
|
|
// Clang-7 as host compiler under nvcc returns types from std::make_unsigned
|
|
// that are not compatible with the vtkm::Atomic API, so we define our own
|
|
// mapping. This must exist for every entry in vtkm::cont::AtomicArrayTypeList.
|
|
template <typename>
|
|
struct MakeUnsigned;
|
|
template <>
|
|
struct MakeUnsigned<vtkm::UInt32>
|
|
{
|
|
using type = vtkm::UInt32;
|
|
};
|
|
template <>
|
|
struct MakeUnsigned<vtkm::Int32>
|
|
{
|
|
using type = vtkm::UInt32;
|
|
};
|
|
template <>
|
|
struct MakeUnsigned<vtkm::UInt64>
|
|
{
|
|
using type = vtkm::UInt64;
|
|
};
|
|
template <>
|
|
struct MakeUnsigned<vtkm::Int64>
|
|
{
|
|
using type = vtkm::UInt64;
|
|
};
|
|
template <>
|
|
struct MakeUnsigned<vtkm::Float32>
|
|
{
|
|
using type = vtkm::UInt32;
|
|
};
|
|
template <>
|
|
struct MakeUnsigned<vtkm::Float64>
|
|
{
|
|
using type = vtkm::UInt64;
|
|
};
|
|
|
|
template <typename T>
|
|
struct ArithType
|
|
{
|
|
using type = typename MakeUnsigned<T>::type;
|
|
};
|
|
template <>
|
|
struct ArithType<vtkm::Float32>
|
|
{
|
|
using type = vtkm::Float32;
|
|
};
|
|
template <>
|
|
struct ArithType<vtkm::Float64>
|
|
{
|
|
using type = vtkm::Float64;
|
|
};
|
|
}
|
|
|
|
template <typename T, typename... MaybeDevice>
|
|
class AtomicArrayExecutionObject;
|
|
|
|
template <typename T, typename Device>
|
|
class VTKM_DEPRECATED(1.6, "AtomicArrayExecutionObject no longer uses Device template parameter.")
|
|
AtomicArrayExecutionObject<T, Device> : public AtomicArrayExecutionObject<T>
|
|
{
|
|
using Superclass = AtomicArrayExecutionObject<T>;
|
|
|
|
public:
|
|
AtomicArrayExecutionObject() = default;
|
|
|
|
// This constructor is deprecated in VTK-m 1.6.
|
|
VTKM_DEPRECATED(1.6, "AtomicArrayExecutionObject no longer uses Device template parameter.")
|
|
AtomicArrayExecutionObject(vtkm::cont::ArrayHandle<T> handle)
|
|
: Superclass(handle, Device{})
|
|
{
|
|
}
|
|
|
|
VTKM_DEPRECATED(1.6, "AtomicArrayExecutionObject no longer uses Device template parameter.")
|
|
AtomicArrayExecutionObject(vtkm::cont::ArrayHandle<T> handle, vtkm::cont::Token& token)
|
|
: Superclass(handle, Device{}, token)
|
|
{
|
|
}
|
|
|
|
// How does this even work?
|
|
template <typename PortalType>
|
|
#ifndef VTKM_MSVC
|
|
// Some versions of visual studio seem to have a bug that causes an error with the
|
|
// deprecated attribute at this location.
|
|
VTKM_DEPRECATED(1.6, "AtomicArrayExecutionObject no longer uses Device template parameter.")
|
|
#endif
|
|
AtomicArrayExecutionObject(const PortalType& portal)
|
|
: Superclass(portal)
|
|
{
|
|
}
|
|
};
|
|
|
|
template <typename T>
|
|
class AtomicArrayExecutionObject<T>
|
|
{
|
|
// Checks if PortalType has a GetIteratorBegin() method that returns a
|
|
// pointer.
|
|
template <typename PortalType,
|
|
typename PointerType = decltype(std::declval<PortalType>().GetIteratorBegin())>
|
|
struct HasPointerAccess : public std::is_pointer<PointerType>
|
|
{
|
|
};
|
|
|
|
public:
|
|
using ValueType = T;
|
|
|
|
AtomicArrayExecutionObject() = default;
|
|
|
|
// This constructor is deprecated in VTK-m 1.6.
|
|
VTKM_CONT VTKM_DEPRECATED(1.6, "AtomicArrayExecutionObject constructor needs token.")
|
|
AtomicArrayExecutionObject(vtkm::cont::ArrayHandle<T> handle,
|
|
vtkm::cont::DeviceAdapterId device)
|
|
: Data{ handle.PrepareForInPlace(device).GetIteratorBegin() }
|
|
, NumberOfValues{ handle.GetNumberOfValues() }
|
|
{
|
|
using PortalType = decltype(handle.PrepareForInPlace(device));
|
|
VTKM_STATIC_ASSERT_MSG(HasPointerAccess<PortalType>::value,
|
|
"Source portal must return a pointer from "
|
|
"GetIteratorBegin().");
|
|
}
|
|
|
|
VTKM_CONT AtomicArrayExecutionObject(vtkm::cont::ArrayHandle<T> handle,
|
|
vtkm::cont::DeviceAdapterId device,
|
|
vtkm::cont::Token& token)
|
|
: Data{ handle.PrepareForInPlace(device, token).GetIteratorBegin() }
|
|
, NumberOfValues{ handle.GetNumberOfValues() }
|
|
{
|
|
using PortalType = decltype(handle.PrepareForInPlace(device, token));
|
|
VTKM_STATIC_ASSERT_MSG(HasPointerAccess<PortalType>::value,
|
|
"Source portal must return a pointer from "
|
|
"GetIteratorBegin().");
|
|
}
|
|
|
|
VTKM_SUPPRESS_EXEC_WARNINGS
|
|
VTKM_EXEC
|
|
vtkm::Id GetNumberOfValues() const { return this->NumberOfValues; }
|
|
|
|
/// \brief Perform an atomic load of the indexed element with acquire memory
|
|
/// ordering.
|
|
/// \param index The index of the element to load.
|
|
/// \return The value of the atomic array at \a index.
|
|
///
|
|
VTKM_SUPPRESS_EXEC_WARNINGS
|
|
VTKM_EXEC
|
|
ValueType Get(vtkm::Id index) const
|
|
{
|
|
// We only support 32/64 bit signed/unsigned ints, and vtkm::Atomic
|
|
// currently only provides API for unsigned types.
|
|
// We'll cast the signed types to unsigned to work around this.
|
|
using APIType = typename detail::MakeUnsigned<ValueType>::type;
|
|
|
|
return static_cast<T>(vtkm::AtomicLoad(reinterpret_cast<APIType*>(this->Data + index)));
|
|
}
|
|
|
|
/// \brief Peform an atomic addition with sequentially consistent memory
|
|
/// ordering.
|
|
/// \param index The index of the array element that will be added to.
|
|
/// \param value The addend of the atomic add operation.
|
|
/// \return The original value of the element at \a index (before addition).
|
|
/// \warning Overflow behavior from this operation is undefined.
|
|
///
|
|
VTKM_SUPPRESS_EXEC_WARNINGS
|
|
VTKM_EXEC
|
|
ValueType Add(vtkm::Id index, const ValueType& value) const
|
|
{
|
|
// We only support 32/64 bit signed/unsigned ints, and vtkm::Atomic
|
|
// currently only provides API for unsigned types.
|
|
// We'll cast the signed types to unsigned to work around this.
|
|
// This is safe, since the only difference between signed/unsigned types
|
|
// is how overflow works, and signed overflow is already undefined. We also
|
|
// document that overflow is undefined for this operation.
|
|
using APIType = typename detail::ArithType<ValueType>::type;
|
|
|
|
return static_cast<T>(
|
|
vtkm::AtomicAdd(reinterpret_cast<APIType*>(this->Data + index), static_cast<APIType>(value)));
|
|
}
|
|
|
|
/// \brief Peform an atomic store to memory while enforcing, at minimum, "release"
|
|
/// memory ordering.
|
|
/// \param index The index of the array element that will be added to.
|
|
/// \param value The value to write for the atomic store operation.
|
|
/// \warning Using something like:
|
|
/// ```
|
|
/// Set(index, Get(index)+N)
|
|
/// ```
|
|
/// Should not be done as it is not thread safe, instead you should use
|
|
/// the provided Add method instead.
|
|
///
|
|
VTKM_SUPPRESS_EXEC_WARNINGS
|
|
VTKM_EXEC
|
|
void Set(vtkm::Id index, const ValueType& value) const
|
|
{
|
|
// We only support 32/64 bit signed/unsigned ints, and vtkm::Atomic
|
|
// currently only provides API for unsigned types.
|
|
// We'll cast the signed types to unsigned to work around this.
|
|
// This is safe, since the only difference between signed/unsigned types
|
|
// is how overflow works, and signed overflow is already undefined. We also
|
|
// document that overflow is undefined for this operation.
|
|
using APIType = typename detail::MakeUnsigned<ValueType>::type;
|
|
|
|
vtkm::AtomicStore(reinterpret_cast<APIType*>(this->Data + index), static_cast<APIType>(value));
|
|
}
|
|
|
|
/// \brief Perform an atomic compare and exchange operation with sequentially consistent
|
|
/// memory ordering.
|
|
/// \param index The index of the array element that will be atomically
|
|
/// modified.
|
|
/// \param oldValue A pointer to the expected value of the indexed element.
|
|
/// \param newValue The value to replace the indexed element with.
|
|
/// \return If the operation is successful, \a true is returned. Otherwise,
|
|
/// \a oldValue is replaced with the current value of the indexed element,
|
|
/// the element is not modified, and \a false is returned. In either case, \a oldValue
|
|
/// becomes the value that was originally in the indexed element.
|
|
///
|
|
/// This operation is typically used in a loop. For example usage,
|
|
/// an atomic multiplication may be implemented using compare-exchange as follows:
|
|
///
|
|
/// ```cpp
|
|
/// AtomicArrayExecutionObject<vtkm::Int32> atomicArray = ...;
|
|
///
|
|
/// // Compare-exchange multiplication:
|
|
/// vtkm::Int32 current = atomicArray.Get(idx); // Load the current value at idx
|
|
/// vtkm::Int32 newVal;
|
|
/// do {
|
|
/// newVal = current * multFactor; // the actual multiplication
|
|
/// } while (!atomicArray.CompareExchange(idx, ¤t, newVal));
|
|
/// ```
|
|
///
|
|
/// The while condition here updates \a newVal what the proper multiplication
|
|
/// is given the expected current value. It then compares this to the
|
|
/// value in the array. If the values match, the operation was successful and the
|
|
/// loop exits. If the values do not match, the value at \a idx was changed
|
|
/// by another thread since the initial Get, and the compare-exchange operation failed --
|
|
/// the target element was not modified by the compare-exchange call. If this happens, the
|
|
/// loop body re-executes using the new value of \a current and tries again until
|
|
/// it succeeds.
|
|
///
|
|
/// Note that for demonstration purposes, the previous code is unnecessarily verbose.
|
|
/// We can express the same atomic operation more succinctly with just two lines where
|
|
/// \a newVal is just computed in place.
|
|
///
|
|
/// ```cpp
|
|
/// vtkm::Int32 current = atomicArray.Get(idx); // Load the current value at idx
|
|
/// while (!atomicArray.CompareExchange(idx, ¤t, current * multFactor));
|
|
/// ```
|
|
///
|
|
VTKM_SUPPRESS_EXEC_WARNINGS
|
|
VTKM_EXEC
|
|
bool CompareExchange(vtkm::Id index, ValueType* oldValue, const ValueType& newValue) const
|
|
{
|
|
// We only support 32/64 bit signed/unsigned ints, and vtkm::Atomic
|
|
// currently only provides API for unsigned types.
|
|
// We'll cast the signed types to unsigned to work around this.
|
|
// This is safe, since the only difference between signed/unsigned types
|
|
// is how overflow works, and signed overflow is already undefined.
|
|
using APIType = typename detail::MakeUnsigned<ValueType>::type;
|
|
|
|
return vtkm::AtomicCompareExchange(reinterpret_cast<APIType*>(this->Data + index),
|
|
reinterpret_cast<APIType*>(oldValue),
|
|
static_cast<APIType>(newValue));
|
|
}
|
|
|
|
VTKM_DEPRECATED(1.6, "Use CompareExchange. (Note the changed interface.)")
|
|
VTKM_EXEC ValueType CompareAndSwap(vtkm::Id index,
|
|
const ValueType& newValue,
|
|
ValueType oldValue) const
|
|
{
|
|
this->CompareExchange(index, &oldValue, newValue);
|
|
return oldValue;
|
|
}
|
|
|
|
private:
|
|
ValueType* Data{ nullptr };
|
|
vtkm::Id NumberOfValues{ 0 };
|
|
};
|
|
}
|
|
} // namespace vtkm::exec
|
|
|
|
#endif //vtk_m_exec_AtomicArrayExecutionObject_h
|