redgpu/ispc: Intel® Implicit SPMD Program Compiler

tests/reduce-add-float16.ispc

12 строк386 B

Arina Neshlyaeva

FP16: reduce_add

03 май 2022, 00:11

03 май 2022, 00:1193c683c

#include "../test_static.isph"
// rule: skip on arch=x86
// rule: skip on arch=x86-64
task void f_fu(uniform float RET[], uniform float aFOO[], uniform float b) {
    float16 v = aFOO[programIndex];
    uniform float m;
    if (v < 3)
        m = reduce_add(-v);
    RET[programIndex] = m;
}

task void result(uniform float RET[]) { RET[programIndex] = (programCount == 1) ? -1 : -3; }