Function example_dynamic_shared_memory

example/extra/example_temporary_storage.cpp:200–223

template<const unsigned int BlockSize, class T>
__global__ __launch_bounds__(BlockSize)
void example_dynamic_shared_memory(const T* input, T* output)
{
    // Indexing for  this block
    unsigned int index = (blockIdx.x * BlockSize) + threadIdx.x;

    // Initialize primitives
    using block_scan_type = rocprim::block_scan<T, BlockSize>;

    // Allocation done in runtime, for more information please visit:
    // https://github.com/ROCm-Developer-Tools/HIP/tree/master/samples/2_Cookbook/6_dynamic_shared
    HIP_DYNAMIC_SHARED(typename block_scan_type::storage_type, primitive_storage);

    // Variables required for performing a scan
    T input_value, output_value;

    // execute inclusive scan
    input_value = input[index];
    block_scan_type().inclusive_scan(input_value,
                                     output_value,
                                     *primitive_storage,
                                     rocprim::plus<T>());

    output[index] = output_value;
}

// Host function that runs example_dynamic_shared_memory kernel
template<class T>