mul.comp 287 B

1234567891011121314
  1. #version 450
  2. #include "types.comp"
  3. #include "generic_binary_head.comp"
  4. void main() {
  5. const uint idx = get_idx();
  6. if (idx >= p.ne) {
  7. return;
  8. }
  9. data_d[p.d_offset + dst_idx(idx)] = D_TYPE(FLOAT_TYPE(data_a[src0_idx(idx)]) * FLOAT_TYPE(data_b[src1_idx(idx)]));
  10. }