redgpu/ispc: Intel® Implicit SPMD Program Compiler

examples/cpu/simple/simple.ispc

24 строки633 B

Dmitry Babokin

01 апр 2023, 03:48

01 апр 2023, 03:4833bd2b4

/*
  Copyright (c) 2010-2023, Intel Corporation

  SPDX-License-Identifier: BSD-3-Clause
*/


export void simple(uniform float vin[], uniform float vout[],
                   uniform int count) {
    foreach (index = 0 ... count) {
        // Load the appropriate input value for this program instance.
        float v = vin[index];

        // Do an arbitrary little computation, but at least make the
        // computation dependent on the value being processed
        if (v < 3.)
            v = v * v;
        else
            v = sqrt(v);

        // And write the result to the output array.
        vout[index] = v;
    }
}