atomicMaxPair.h File Reference

#include <alpaka/alpaka.hpp>
#include "FWCore/Utilities/interface/bit_cast.h"
#include "HeterogeneousCore/AlpakaInterface/interface/config.h"

Go to the source code of this file.

Functions
template<typename TAcc , typename = std::enable_if_t<alpaka::isAccelerator<TAcc>>, typename F >
ALPAKA_FN_HOST_ACC ALPAKA_FN_INLINE void	atomicMaxPair (const TAcc &acc, unsigned long long int *address, std::pair< unsigned int, float > value, F comparator)

Function Documentation

atomicMaxPair()

template<typename TAcc , typename = std::enable_if_t<alpaka::isAccelerator<TAcc>>, typename F >

ALPAKA_FN_HOST_ACC ALPAKA_FN_INLINE void atomicMaxPair	(	const TAcc &	acc,
		unsigned long long int *	address,
		std::pair< unsigned int, float >	value,
		F	comparator
	)

Definition at line 10 of file atomicMaxPair.h.

References cms::cudacompat::atomicCAS(), runTheMatrix::ret, and heppy_batch::val.

Referenced by ALPAKA_ACCELERATOR_NAMESPACE::hcal::reconstruction::mahi::Kernel_prep1d_sameNumberOfSamples::operator()().

                                                                     {
#if defined(__CUDA_ARCH__) or defined(__HIP_DEVICE_COMPILE__)
  unsigned long long int val = (static_cast<unsigned long long int>(value.first) << 32) + __float_as_uint(value.second);
  unsigned long long int ret = *address;
  while (comparator(value,
                    std::pair<unsigned int, float>{static_cast<unsigned int>(ret >> 32 & 0xffffffff),
                                                   __uint_as_float(ret & 0xffffffff)})) {
    unsigned long long int old = ret;
    if ((ret = atomicCAS(address, old, val)) == old)
      break;
  }
#else
  unsigned long long int val =
      (static_cast<unsigned long long int>(value.first) << 32) + edm::bit_cast<unsigned int>(value.second);
  unsigned long long int ret = *address;
  while (comparator(value,
                    std::pair{static_cast<unsigned int>(ret >> 32 & 0xffffffff),
                              edm::bit_cast<float>(static_cast<unsigned int>(ret & 0xffffffff))})) {
    unsigned long long int old = ret;
    if ((ret = alpaka::atomicCas(acc, address, old, val)) == old)
      break;
  }
#endif  // __CUDA_ARCH__ or __HIP_DEVICE_COMPILE__
}