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
|
#version 460 core
#extension GL_ARM_tensors : enable
#extension GL_EXT_shader_explicit_arithmetic_types : enable
// Int types
uniform tensorARM<int8_t, 1> it8;
uniform tensorARM<int16_t, 1> it16;
uniform tensorARM<int, 1> it32;
uniform tensorARM<int64_t, 1> it64;
// Uint types
uniform tensorARM<uint8_t, 1> ut8;
uniform tensorARM<uint16_t, 1> ut16;
uniform tensorARM<uint, 1> ut32;
uniform tensorARM<uint64_t, 1> ut64;
// Float types
uniform tensorARM<float16_t, 1> ft16;
uniform tensorARM<float, 1> ft32;
uniform tensorARM<float64_t, 1> ft64;
void main() {
uint coords[] = {0};
// Int types
int8_t iw8 = int8_t(1);
int16_t iw16 = 1s;
int iw = 1;
int64_t iw64 = 1l;
tensorWriteARM(it8, coords, iw8);
tensorWriteARM(it16, coords, iw16);
tensorWriteARM(it32, coords, iw);
tensorWriteARM(it64, coords, iw64);
int iwv[4] = int[](1, 1, 1, 1);
tensorWriteARM(it32, coords, iwv);
int8_t ir8;
int16_t ir16;
int ir;
int64_t ir64;
tensorReadARM(it8, coords, ir8);
tensorReadARM(it16, coords, ir16);
tensorReadARM(it32, coords, ir);
tensorReadARM(it64, coords, ir64);
int irv[4];
tensorReadARM(it32, coords, irv);
// Uint types
uint8_t uw8 = uint8_t(1);
uint16_t uw16 = 1s;
uint uw = 1;
uint64_t uw64 = 1l;
tensorWriteARM(ut8, coords, uw8);
tensorWriteARM(ut16, coords, uw16);
tensorWriteARM(ut32, coords, uw);
tensorWriteARM(ut64, coords, uw64);
uint uwv[4] = uint[](1, 1, 1, 1);
tensorWriteARM(ut32, coords, uwv);
uint8_t ur8;
uint16_t ur16;
uint ur;
uint64_t ur64;
tensorReadARM(ut8, coords, ur8);
tensorReadARM(ut16, coords, ur16);
tensorReadARM(ut32, coords, ur);
tensorReadARM(ut64, coords, ur64);
uint urv[4];
tensorReadARM(ut32, coords, urv);
// Float types
float fw = 1.0;
float16_t fw16 = 1.0hf;
float64_t fw64 = 1.0;
tensorWriteARM(ft16, coords, fw16);
tensorWriteARM(ft32, coords, fw);
tensorWriteARM(ft64, coords, fw64);
float fwv[4] = float[](1.0, 1.0, 1.0, 1.0);
tensorWriteARM(ft32, coords, fwv);
float fr;
float16_t fr16;
float64_t fr64;
tensorReadARM(ft16, coords, fr16);
tensorReadARM(ft32, coords, fr);
tensorReadARM(ft64, coords, fr64);
float frv[4];
tensorReadARM(ft32, coords, frv);
}
|
