docs/api/launch__bounds_8h_source.html

#ifndef LAUNCH_BOUNDS_H

#define LAUNCH_BOUNDS_H


#include <Kokkos_Core.hpp>


#if defined(KOKKOS_ENABLE_HIP)

using HeavyBounds = Kokkos::LaunchBounds<256, 1>;

#else

using HeavyBounds = Kokkos::LaunchBounds<>;

#endif


template <typename... Traits>

using HeavyRange = Kokkos::RangePolicy<HeavyBounds, Traits...>;


#endif // LAUNCH_BOUNDS_H

dot
KOKKOS_INLINE_FUNCTION rtype dot(const rtype *a, const rtype *b)
Compute the dot product of two arrays.
Definition common_math.h:32

HeavyBounds
Kokkos::LaunchBounds<> HeavyBounds
HIP compiles a kernel without launch bounds for blocks of 1024 threads, which caps it at 128 register...
Definition launch_bounds.h:22

HeavyRange
Kokkos::RangePolicy< HeavyBounds, Traits... > HeavyRange
Range policy for register-heavy kernels.
Definition launch_bounds.h:27