xref: /aosp_15_r20/external/skia/tests/sksl/compute/MatrixMultiply.wgsl (revision c8dee2aa9b3f27cf6c858bd81872bdeb2c07ed17)

diagnostic(off, derivative_uniformity);
diagnostic(off, chromium.unreachable_code);
struct CSIn {
  @builtin(global_invocation_id) sk_GlobalInvocationID: vec3<u32>,
};
struct sizeBuffer {
  sizes: array<vec2<i32>>,
};
@group(0) @binding(0) var<storage, read_write> _storage0 : sizeBuffer;
struct inputs1 {
  data1: array<f32>,
};
@group(0) @binding(1) var<storage, read> _storage1 : inputs1;
struct inputs2 {
  data2: array<f32>,
};
@group(0) @binding(2) var<storage, read> _storage2 : inputs2;
struct result {
  resultData: array<f32>,
};
@group(0) @binding(3) var<storage, read_write> _storage3 : result;
fn _skslMain(_stageIn: CSIn) {
  {
    _storage0.sizes[2] = vec2<i32>(_storage0.sizes[0].x, _storage0.sizes[1].y);
    let resultCell: vec2<i32> = vec2<i32>(i32(_stageIn.sk_GlobalInvocationID.x), i32(_stageIn.sk_GlobalInvocationID.y));
    var result: f32 = 0.0;
    {
      var i: i32 = 0;
      loop {
        if i < _storage0.sizes[0].y {
          {
            let a: i32 = i + resultCell.x * _storage0.sizes[0].y;
            let b: i32 = resultCell.y + i * _storage0.sizes[1].y;
            result = result + _storage1.data1[a] * _storage2.data2[b];
          }
        } else {
          break;
        }
        continuing {
          i = i + i32(1);
        }
      }
    }
    let index: i32 = resultCell.y + resultCell.x * _storage0.sizes[1].y;
    _storage3.resultData[index] = result;
  }
}
@compute @workgroup_size(16, 16, 1) fn main(_stageIn: CSIn) {
  _skslMain(_stageIn);
}