Skip to content

Commit 3254fd3

Browse files
authored
[BE][ET-VK] update full to use layout gen
Differential Revision: D71825477 Pull Request resolved: #9590
1 parent 8510996 commit 3254fd3

File tree

2 files changed

+5
-10
lines changed

2 files changed

+5
-10
lines changed

backends/vulkan/runtime/graph/ops/glsl/full.glsl

Lines changed: 4 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -18,15 +18,9 @@
1818

1919
layout(std430) buffer;
2020

21-
layout(set = 0, binding = 0, ${IMAGE_FORMAT[DTYPE]}) uniform PRECISION restrict writeonly ${IMAGE_T[NDIM][DTYPE]} image_out;
22-
23-
layout(set = 0, binding = 1) uniform PRECISION restrict Sizes {
24-
ivec4 sizes;
25-
};
26-
27-
layout(set = 0, binding = 2) uniform PRECISION restrict FillVal {
28-
float fill_value;
29-
};
21+
${layout_declare_tensor(B, "w", "t_out", DTYPE, STORAGE)}
22+
${layout_declare_ubo(B, "ivec4", "sizes")}
23+
${layout_declare_ubo(B, "float", "fill_value")}
3024

3125
layout(local_size_x_id = 0, local_size_y_id = 1, local_size_z_id = 2) in;
3226

@@ -50,5 +44,5 @@ void main() {
5044
outtex = outtex * valid_idx;
5145
}
5246

53-
imageStore(image_out, POS, outtex);
47+
imageStore(t_out, POS, outtex);
5448
}

backends/vulkan/runtime/graph/ops/glsl/full.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@ full:
99
NDIM: 3
1010
DTYPE: float
1111
PACKING: C_packed
12+
STORAGE: texture3d
1213
generate_variant_forall:
1314
DTYPE:
1415
- VALUE: half

0 commit comments

Comments
 (0)