Compiler Explorer

Source code

export uniform float dot_product(const float* uniform a, const float* uniform b, uniform size_t size)
{
    varying float acc = 0.0f;

size = size & ~31; // let the compiler know size is multiple of 32
    foreach(i = 0...size)
    {
        varying float suba = a[i];
        varying float subb = b[i];
        acc += suba * subb;
    }

return reduce_add(acc);
}

export uniform float dot_product_multiple_acc(const float* uniform a, const float* uniform b, uniform size_t size)
{
    varying float acc1 = 0.0f;
    varying float acc2 = 0.0f;

size = size & ~31; // let the compiler know size is multiple of 32
    for (uniform size_t base = 0; base < size; base += programCount * 2) {
        varying size_t i1 = base + programIndex;
        varying size_t i2 = base + programCount + programIndex;
        varying float suba1 = a[i1];
        varying float subb1 = b[i1];
        varying float suba2 = a[i2];
        varying float subb2 = b[i2];
        acc1 += suba1 * subb1;
        acc2 += suba2 * subb2;
    }

return reduce_add(acc1 + acc2);
}