Commit ·
94b2840
1
Parent(s): 49dcffa
Re-export the Vulkan fp16 variant with rank-3 attention
Browse filesAttention 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:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8f9aed61f17b26a45551224ba06f6268431d476edbcc71abb43d4e70a22c23bf
|
| 3 |
+
size 45208066
|