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
|
export const description = `
Execution tests for the 'quantizeToF16' builtin function
T is f32 or vecN<f32>
@const fn quantizeToF16(e: T ) -> T
Quantizes a 32-bit floating point value e as if e were converted to a IEEE 754
binary16 value, and then converted back to a IEEE 754 binary32 value.
Component-wise when T is a vector.
`;
import { makeTestGroup } from '../../../../../../common/framework/test_group.js';
import { GPUTest } from '../../../../../gpu_test.js';
import { kValue } from '../../../../../util/constants.js';
import { TypeF32 } from '../../../../../util/conversion.js';
import { quantizeToF16Interval } from '../../../../../util/f32_interval.js';
import { fullF16Range, fullF32Range } from '../../../../../util/math.js';
import { makeCaseCache } from '../../case_cache.js';
import { allInputSources, generateUnaryToF32IntervalCases, run } from '../../expression.js';
import { builtin } from './builtin.js';
export const g = makeTestGroup(GPUTest);
export const d = makeCaseCache('quantizeToF16', {
f32_const: () => {
return generateUnaryToF32IntervalCases(
[
kValue.f16.negative.min,
kValue.f16.negative.max,
kValue.f16.subnormal.negative.min,
kValue.f16.subnormal.negative.max,
kValue.f16.subnormal.positive.min,
kValue.f16.subnormal.positive.max,
kValue.f16.positive.min,
kValue.f16.positive.max,
...fullF16Range(),
],
'f32-only',
quantizeToF16Interval
);
},
f32_non_const: () => {
return generateUnaryToF32IntervalCases(
[
kValue.f16.negative.min,
kValue.f16.negative.max,
kValue.f16.subnormal.negative.min,
kValue.f16.subnormal.negative.max,
kValue.f16.subnormal.positive.min,
kValue.f16.subnormal.positive.max,
kValue.f16.positive.min,
kValue.f16.positive.max,
...fullF32Range(),
],
'unfiltered',
quantizeToF16Interval
);
},
});
g.test('f32')
.specURL('https://www.w3.org/TR/WGSL/#float-builtin-functions')
.desc(`f32 tests`)
.params(u =>
u.combine('inputSource', allInputSources).combine('vectorize', [undefined, 2, 3, 4] as const)
)
.fn(async t => {
const cases = await d.get(t.params.inputSource === 'const' ? 'f32_const' : 'f32_non_const');
await run(t, builtin('quantizeToF16'), [TypeF32], TypeF32, t.params, cases);
});
|