yum-mirror/glslang

git clone https://git.yummers.dev/yum-mirror/glslang

Jeff BolzImplement GL_EXT_ocp_microscaling_types62f5430e

main
branchesmainall refs →
2.4 KiB77 linesraw
1#version 450 core
2
3#extension GL_EXT_bfloat16 : require
4#extension GL_EXT_float_mxint8 : require
5#extension GL_KHR_cooperative_matrix : enable
6#extension GL_NV_cooperative_matrix2 : enable
7#extension GL_KHR_memory_scope_semantics : enable
8#extension GL_EXT_shader_explicit_arithmetic_types : enable
9#extension GL_EXT_scalar_block_layout : enable
10#extension GL_EXT_long_vector : enable
11
12layout(local_size_x = 1, local_size_y = 1, local_size_z = 1) in;
13
14const floatmxint8_t bc10 = floatmxint8_t(10);
15const fmxint8vec2 b2c20 = fmxint8vec2(20);
16layout(constant_id = 1) const floatmxint8_t bsc10 = floatmxint8_t(10);
17
18struct S { floatmxint8_t b; };
19const S sc = S(bc10);
20
21const floatmxint8_t bca[2] = {bc10, bc10};
22
23shared floatmxint8_t bfs[10];
24
25layout(scalar) buffer B {
26    fmxint8vec3 b3;
27    fmxint8vec2 b2;
28    floatmxint8_t b1;
29} buf;
30
31floatmxint8_t funcfmxint8(floatmxint8_t x)
32{
33    return x;
34}
35
36void main()
37{
38    float f = 2.0;
39    floatmxint8_t b = floatmxint8_t(1.0fmxint8);
40
41    float16_t f16 = float16_t(7);
42    float32_t f32 = 7;
43    float64_t f64 = 7;
44    f16 = float16_t(b);
45    f32 = float32_t(b);
46    f64 = float64_t(b);
47
48    b = funcfmxint8(b);
49
50    b = buf.b1;
51    fmxint8vec2 b2 = buf.b2;
52    fmxint8vec3 b3 = buf.b3;
53
54    int8_t i8_1 = floatmxint8BitsToIntEXT(b);
55    i8vec2 i8_2 = floatmxint8BitsToIntEXT(b2);
56    i8vec3 i8_3 = floatmxint8BitsToIntEXT(fmxint8vec3(b2, b));
57    i8vec4 i8_4 = floatmxint8BitsToIntEXT(fmxint8vec4(b2, b2));
58    vector<int8_t, 5> i8_5 = floatmxint8BitsToIntEXT(vector<floatmxint8_t, 5>(b2, b2, b));
59
60    uint8_t u8_1 = floatmxint8BitsToUintEXT(b);
61    u8vec2  u8_2 = floatmxint8BitsToUintEXT(b2);
62    u8vec3  u8_3 = floatmxint8BitsToUintEXT(fmxint8vec3(b2, b));
63    u8vec4  u8_4 = floatmxint8BitsToUintEXT(fmxint8vec4(b2, b2));
64    vector<uint8_t, 5> u8_5 = floatmxint8BitsToUintEXT(vector<floatmxint8_t, 5>(b2, b2, b));
65
66    floatmxint8_t f8_1 = intBitsToFloatmxint8EXT(i8_1);
67    fmxint8vec2   f8_2 = intBitsToFloatmxint8EXT(i8_2);
68    fmxint8vec3   f8_3 = intBitsToFloatmxint8EXT(i8_3);
69    fmxint8vec4   f8_4 = intBitsToFloatmxint8EXT(i8_4);
70    vector<floatmxint8_t, 5> f8_5 = intBitsToFloatmxint8EXT(i8_5);
71
72    f8_1 = uintBitsToFloatmxint8EXT(u8_1);
73    f8_2 = uintBitsToFloatmxint8EXT(u8_2);
74    f8_3 = uintBitsToFloatmxint8EXT(u8_3);
75    f8_4 = uintBitsToFloatmxint8EXT(u8_4);
76    f8_5 = uintBitsToFloatmxint8EXT(u8_5);
77}