fill_kernel.hip.hpp

Go to the documentation of this file.

 
#ifndef PLSSVM_BACKENDS_HIP_DETAIL_FILL_KERNEL_HPP_
#define PLSSVM_BACKENDS_HIP_DETAIL_FILL_KERNEL_HPP_
#pragma once
 
#include "hip/hip_runtime.h"      // HIP runtime functions
#include "hip/hip_runtime_api.h"  // HIP runtime functions
 
namespace plssvm::hip::detail {
 
template <typename value_type, typename size_type>
__global__ void fill_array(value_type *data, value_type value, size_type pos, size_type count) {
    const unsigned int idx = blockIdx.x * blockDim.x + threadIdx.x;
    // fill the array
    if (idx < count) {
        data[pos + idx] = value;
    }
}
 
}  // namespace plssvm::hip::detail
 
#endif  // PLSSVM_BACKENDS_HIP_DETAIL_FILL_KERNEL_HPP_