mirror of https://github.com/xemu-project/xemu.git
31 lines
684 B
Plaintext
31 lines
684 B
Plaintext
#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);
|
|
}
|