cuda.fill.FillTwoFloatArraysKernel.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 fillTwoFloatsArraysKernel(
    int numberRows,
    int numberEntries,
    float* firstArray,
    float firstConstant,
    float* secondArray,
    float secondConstant) {

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

    firstArray[index] = firstConstant;
    secondArray[index] = secondConstant;

}