1  #version 450
 2  
 3  #include "types.comp"
 4  #include "generic_unary_head.comp"
 5  
 6  void main() {
 7      if (gl_GlobalInvocationID.x >= p.ne) {
 8          return;
 9      }
10  
11      const FLOAT_TYPE val = FLOAT_TYPE(data_a[src0_idx(gl_GlobalInvocationID.x)]);
12      data_d[p.d_offset + dst_idx(gl_GlobalInvocationID.x)] = D_TYPE(val * val);
13  }