Files
UnrealEngine/Engine/Source/ThirdParty/Intel/ISPC/ispc-1.24.0/tests/uniform-1.ispc
2025-05-18 13:04:45 +08:00

36 lines
984 B
Plaintext

#include "../test_static.isph"
// rule: skip on arch=xe64
#define THREADS 4
task void f_f_task(uniform float RET[], uniform float aFOO[]) {
uniform float val[programCount];
for (uniform int i = 0; i < programCount; ++i)
val[i] = 0;
foreach (i = 0 ... programCount)
val[i] += aFOO[programCount*taskIndex + i] - 1;
uniform float sum = 0;
for (uniform int i = 0; i < programCount; ++i)
sum += val[i];
if (programIndex < programCount/THREADS)
RET[programCount/THREADS*taskIndex + programIndex] = sum;
}
task void f_f(uniform float RET[], uniform float aFOO[])
{
launch[THREADS] f_f_task(RET, aFOO);
}
task void result_task(uniform float RET[])
{
const uniform float ret = reduce_add(programIndex + programCount*taskIndex);
if (programIndex < programCount/THREADS)
RET[programCount/THREADS*taskIndex + programIndex] = ret;
}
task void result(uniform float RET[]) {
launch[THREADS] result_task(RET);
}