cuda.fill.FillOneFloatArrayKernel.cu Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of komputation Show documentation

Komputation is a neural network framework for the JVM written in the Kotlin programming language.

There is a newer version: 0.12.5

__global__ void fillOneFloatArrayKernel(
    int numberRows,
    int numberEntries,
    float* array,
    float constant) {

    int index = blockIdx.x * numberEntries + blockIdx.y * numberRows + threadIdx.x;

    array[index] = constant;

}