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
|
#version 460 core
#extension GL_ARM_tensors : enable
#extension GL_EXT_shader_explicit_arithmetic_types : enable
layout(set = 0, binding = 0) uniform tensorARM<int32_t, 4> tu;
layout(set = 0, binding = 1) uniform tensorARM<int32_t, 2> td;
layout(set = 0, binding = 2, std430) buffer buff {
uint out_data[];
};
void unusedTensorParam(tensorARM) {
return;
}
uint getDim0Size(tensorARM t) {
return tensorSizeARM(t, 0);
}
uint partiallySpecialized(tensorARM<int> t) {
return tensorSizeARM(t, 1);
}
void main() {
unusedTensorParam(tu);
unusedTensorParam(td);
out_data[0] = getDim0Size(tu);
out_data[0] += getDim0Size(td);
out_data[0] += partiallySpecialized(tu);
}
|
