summaryrefslogtreecommitdiffstats
path: root/dom/webgpu/tests/cts/checkout/src/webgpu/shader/validation/expression/call/builtin/arrayLength.spec.ts
blob: 27d8814b145738e37821e3cdb90ad3941fb4ac8e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
export const description = `
Validation tests for arrayLength builtins.
`;

import { makeTestGroup } from '../../../../../../common/framework/test_group.js';
import { ShaderValidationTest } from '../../../shader_validation_test.js';

export const g = makeTestGroup(ShaderValidationTest);

g.test('bool_type')
  .specURL('https://www.w3.org/TR/WGSL/#arrayLength-builtin')
  .desc(
    `
arrayLength accepts only runtime-sized arrays
`
  )
  .fn(t => {
    const code = `
@compute @workgroup_size(1)
fn main() {
  var b = true;
  _ = arrayLength(&b);
}`;

    t.expectCompileResult(false, code);
  });

const atomic_types = ['u32', 'i32'].map(j => `atomic<${j}>`);
const vec_types = [2, 3, 4]
  .map(i => ['i32', 'u32', 'f32', 'f16'].map(j => `vec${i}<${j}>`))
  .reduce((a, c) => a.concat(c), []);
const f32_matrix_types = [2, 3, 4]
  .map(i => [2, 3, 4].map(j => `mat${i}x${j}f`))
  .reduce((a, c) => a.concat(c), []);
const f16_matrix_types = [2, 3, 4]
  .map(i => [2, 3, 4].map(j => `mat${i}x${j}<f16>`))
  .reduce((a, c) => a.concat(c), []);

g.test('type')
  .specURL('https://www.w3.org/TR/WGSL/#arrayLength-builtin')
  .desc(
    `
arrayLength accepts only runtime-sized arrays
`
  )
  .params(u =>
    u.combine('type', [
      'i32',
      'u32',
      'f32',
      'f16',
      ...f32_matrix_types,
      ...f16_matrix_types,
      ...vec_types,
      ...atomic_types,
      'T',
      'array<i32, 2>',
      'array<i32>',
    ])
  )
  .beforeAllSubcases(t => {
    if (t.params.type.includes('f16')) {
      t.selectDeviceOrSkipTestCase('shader-f16');
    }
  })
  .fn(t => {
    const code = `
struct T {
  b: i32,
}
struct S {
  ary: ${t.params.type}
}

@group(0) @binding(0) var<storage, read_write> items: S;

@compute @workgroup_size(1)
fn main() {
  _ = arrayLength(&items.ary);
}`;

    t.expectCompileResult(t.params.type === 'array<i32>', code);
  });

// Note, the `write` case actually fails because you can't declare a storage buffer of
// access_mode `write`.
g.test('access_mode')
  .specURL('https://www.w3.org/TR/WGSL/#arrayLength-builtin')
  .desc(
    `
arrayLength runtime-sized array must have an access_mode of read or read_write
`
  )
  .params(u => u.combine('mode', ['read', 'read_write', 'write']))
  .fn(t => {
    const code = `
struct S {
  ary: array<i32>,
}

@group(0) @binding(0) var<storage, ${t.params.mode}> items: S;

@compute @workgroup_size(1)
fn main() {
  _ = arrayLength(&items.ary);
}`;

    t.expectCompileResult(t.params.mode !== 'write', code);
  });