xref: /aosp_15_r20/external/skia/tests/sksl/intrinsics/Sqrt.wgsl (revision c8dee2aa9b3f27cf6c858bd81872bdeb2c07ed17)
1diagnostic(off, derivative_uniformity);
2diagnostic(off, chromium.unreachable_code);
3struct FSOut {
4  @location(0) sk_FragColor: vec4<f32>,
5};
6struct _GlobalUniforms {
7  testMatrix2x2: mat2x2<f32>,
8  colorGreen: vec4<f32>,
9  colorRed: vec4<f32>,
10};
11@binding(0) @group(0) var<uniform> _globalUniforms: _GlobalUniforms;
12fn _skslMain(_skParam0: vec2<f32>) -> vec4<f32> {
13  var coords = _skParam0;
14  {
15    const negativeVal: vec4<f32> = vec4<f32>(-1.0, -4.0, -16.0, -64.0);
16    let _skTemp0 = negativeVal;
17    let _skTemp1 = sqrt(_skTemp0);
18    coords = _skTemp1.xy;
19    let inputVal: vec4<f32> = vec4<f32>(_globalUniforms.testMatrix2x2[0], _globalUniforms.testMatrix2x2[1]) + vec4<f32>(0.0, 2.0, 6.0, 12.0);
20    const expected: vec4<f32> = vec4<f32>(1.0, 2.0, 3.0, 4.0);
21    const allowedDelta: vec4<f32> = vec4<f32>(0.05);
22    let _skTemp2 = sqrt(inputVal.x);
23    let _skTemp3 = abs(_skTemp2 - 1.0);
24    let _skTemp4 = sqrt(inputVal.xy);
25    let _skTemp5 = abs(_skTemp4 - vec2<f32>(1.0, 2.0));
26    let _skTemp6 = all((_skTemp5 < vec2<f32>(0.05)));
27    let _skTemp7 = sqrt(inputVal.xyz);
28    let _skTemp8 = abs(_skTemp7 - vec3<f32>(1.0, 2.0, 3.0));
29    let _skTemp9 = all((_skTemp8 < vec3<f32>(0.05)));
30    let _skTemp10 = sqrt(inputVal);
31    let _skTemp11 = abs(_skTemp10 - expected);
32    let _skTemp12 = all((_skTemp11 < allowedDelta));
33    return select(_globalUniforms.colorRed, _globalUniforms.colorGreen, vec4<bool>((((_skTemp3 < 0.05) && _skTemp6) && _skTemp9) && _skTemp12));
34  }
35}
36@fragment fn main() -> FSOut {
37  var _stageOut: FSOut;
38  _stageOut.sk_FragColor = _skslMain(/*fragcoord*/ vec2<f32>());
39  return _stageOut;
40}
41