log.comp 374 B

1234567891011121314151617
  1. #version 450
  2. #include "types.glsl"
  3. #include "generic_unary_head.glsl"
  4. layout(local_size_x = 512, local_size_y = 1, local_size_z = 1) in;
  5. void main() {
  6. const uint idx = get_idx();
  7. if (idx >= p.ne) {
  8. return;
  9. }
  10. const FLOAT_TYPE val = FLOAT_TYPE(data_a[get_aoffset() + src0_idx(idx)]);
  11. data_d[get_doffset() + dst_idx(idx)] = D_TYPE(log(val));
  12. }