Skip to content

Commit 67e01f2

Browse files
committed
Update on "[ET-VK] Implement linear_qcs4w"
## Context Title says it all! ## Changes Extended the implementation of `linear_qcsnw` to support packed 4-bit weight tensors. Differential Revision: [D73941991](https://our.internmc.facebook.com/intern/diff/D73941991/) [ghstack-poisoned]
2 parents d6df8b4 + 23dffed commit 67e01f2

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

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

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,3 +28,11 @@ linear_qcsnw_coop:
2828
OUT_STORAGE: buffer
2929
WEIGHT_STORAGE: buffer
3030
SCALES_STORAGE: buffer
31+
- NAME: linear_qcs4w_coop_texture3d_texture3d_texture2d_texture2d_float
32+
TILE_TXCOLS: 2
33+
QUANT_NBITS: 4
34+
- NAME: linear_qcs4w_coop_buffer_buffer_texture2d_texture2d_float
35+
IN_STORAGE: buffer
36+
OUT_STORAGE: buffer
37+
TILE_TXCOLS: 2
38+
QUANT_NBITS: 4

0 commit comments

Comments
 (0)