aceOfDiamonds commited on
Commit
94b2840
·
1 Parent(s): 49dcffa

Re-export the Vulkan fp16 variant with rank-3 attention

Browse files

Attention now folds the singleton batch into the head dim and runs as
[H, T, D] batched matmuls. The stock 4-D reshapes each become their own
materialised view dispatch on Vulkan; this drops them and is bit-identical to
the previous export on CPU.

At the published 254-token shape: 10.30 -> 8.82 ms on an Adreno 840 and
107.58 -> 97.94 ms on a Mali-G76, which is 2.32x and 3.11x XNNPACK.

vulkan/all_minilm_l6_v2_vulkan_fp16.pte CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:38744f91a7499d5bdf4ba3e9047a1b8a3189f85fcf041dde4cce4eb1783bf4fb
3
- size 45213698
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8f9aed61f17b26a45551224ba06f6268431d476edbcc71abb43d4e70a22c23bf
3
+ size 45208066