Add XNNPACK 8da4w export
Browse files- .gitattributes +1 -0
- README.md +21 -8
- xnnpack/config.json +33 -0
- xnnpack/multi_qa_mpnet_base_dot_v1_xnnpack_8da4w.pte +3 -0
.gitattributes
CHANGED
|
@@ -38,3 +38,4 @@ multi-qa-mpnet-base-dot-v1_xnnpack.pte filter=lfs diff=lfs merge=lfs -text
|
|
| 38 |
xnnpack/multi_qa_mpnet_base_dot_v1_xnnpack_fp32.pte filter=lfs diff=lfs merge=lfs -text
|
| 39 |
vulkan/multi_qa_mpnet_base_dot_v1_vulkan_fp16.pte filter=lfs diff=lfs merge=lfs -text
|
| 40 |
vulkan/multi_qa_mpnet_base_dot_v1_vulkan_int8.pte filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
| 38 |
xnnpack/multi_qa_mpnet_base_dot_v1_xnnpack_fp32.pte filter=lfs diff=lfs merge=lfs -text
|
| 39 |
vulkan/multi_qa_mpnet_base_dot_v1_vulkan_fp16.pte filter=lfs diff=lfs merge=lfs -text
|
| 40 |
vulkan/multi_qa_mpnet_base_dot_v1_vulkan_int8.pte filter=lfs diff=lfs merge=lfs -text
|
| 41 |
+
xnnpack/multi_qa_mpnet_base_dot_v1_xnnpack_8da4w.pte filter=lfs diff=lfs merge=lfs -text
|
README.md
CHANGED
|
@@ -18,19 +18,32 @@ Upstream model: [multi-qa-mpnet-base-dot-v1](https://huggingface.co/sentence-tra
|
|
| 18 |
| --- | --- | --- |
|
| 19 |
| `vulkan/multi_qa_mpnet_base_dot_v1_vulkan_fp16.pte` | vulkan | fp16 |
|
| 20 |
| `vulkan/multi_qa_mpnet_base_dot_v1_vulkan_int8.pte` | vulkan | int8 |
|
|
|
|
| 21 |
| `xnnpack/multi_qa_mpnet_base_dot_v1_xnnpack_fp32.pte` | xnnpack | fp32 |
|
| 22 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 23 |
## Repository structure
|
| 24 |
|
| 25 |
```
|
| 26 |
-
config.json
|
| 27 |
-
tokenizer.json
|
| 28 |
-
tokenizer_config.json
|
| 29 |
-
vulkan/config.json
|
| 30 |
-
vulkan/multi_qa_mpnet_base_dot_v1_vulkan_fp16.pte
|
| 31 |
-
vulkan/multi_qa_mpnet_base_dot_v1_vulkan_int8.pte
|
| 32 |
-
xnnpack/config.json
|
| 33 |
-
xnnpack/
|
|
|
|
| 34 |
```
|
| 35 |
|
| 36 |
## Compatibility
|
|
|
|
| 18 |
| --- | --- | --- |
|
| 19 |
| `vulkan/multi_qa_mpnet_base_dot_v1_vulkan_fp16.pte` | vulkan | fp16 |
|
| 20 |
| `vulkan/multi_qa_mpnet_base_dot_v1_vulkan_int8.pte` | vulkan | int8 |
|
| 21 |
+
| `xnnpack/multi_qa_mpnet_base_dot_v1_xnnpack_8da4w.pte` | xnnpack | 8da4w |
|
| 22 |
| `xnnpack/multi_qa_mpnet_base_dot_v1_xnnpack_fp32.pte` | xnnpack | fp32 |
|
| 23 |
|
| 24 |
+
## XNNPACK int8
|
| 25 |
+
|
| 26 |
+
`xnnpack/multi_qa_mpnet_base_dot_v1_xnnpack_8da4w.pte` is 8da4w (int8 dynamic activations, int4 weights,
|
| 27 |
+
group 32) on the Linear layers, with an int8 word-embedding table. SQuAD v1.1 dev
|
| 28 |
+
passage retrieval (2000 questions against their 315 paragraphs), dot product:
|
| 29 |
+
|
| 30 |
+
| | top-1 | MRR |
|
| 31 |
+
| --- | --- | --- |
|
| 32 |
+
| fp32 | 64.60 | 0.7645 |
|
| 33 |
+
| 8da4w | 63.80 | 0.7578 |
|
| 34 |
+
|
| 35 |
## Repository structure
|
| 36 |
|
| 37 |
```
|
| 38 |
+
config.json 48 B
|
| 39 |
+
tokenizer.json 694 kB
|
| 40 |
+
tokenizer_config.json 363 B
|
| 41 |
+
vulkan/config.json 1.6 kB
|
| 42 |
+
vulkan/multi_qa_mpnet_base_dot_v1_vulkan_fp16.pte 208 MB
|
| 43 |
+
vulkan/multi_qa_mpnet_base_dot_v1_vulkan_int8.pte 127 MB
|
| 44 |
+
xnnpack/config.json 1.6 kB
|
| 45 |
+
xnnpack/multi_qa_mpnet_base_dot_v1_xnnpack_8da4w.pte 69.1 MB
|
| 46 |
+
xnnpack/multi_qa_mpnet_base_dot_v1_xnnpack_fp32.pte 416 MB
|
| 47 |
```
|
| 48 |
|
| 49 |
## Compatibility
|
xnnpack/config.json
CHANGED
|
@@ -40,6 +40,39 @@
|
|
| 40 |
]
|
| 41 |
}
|
| 42 |
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 43 |
}
|
| 44 |
]
|
| 45 |
}
|
|
|
|
| 40 |
]
|
| 41 |
}
|
| 42 |
}
|
| 43 |
+
},
|
| 44 |
+
{
|
| 45 |
+
"file": "multi_qa_mpnet_base_dot_v1_xnnpack_8da4w.pte",
|
| 46 |
+
"precision": "8da4w",
|
| 47 |
+
"methods": {
|
| 48 |
+
"forward": {
|
| 49 |
+
"inputs": [
|
| 50 |
+
{
|
| 51 |
+
"shape": [
|
| 52 |
+
1,
|
| 53 |
+
510
|
| 54 |
+
],
|
| 55 |
+
"dtype": "int64"
|
| 56 |
+
},
|
| 57 |
+
{
|
| 58 |
+
"shape": [
|
| 59 |
+
1,
|
| 60 |
+
510
|
| 61 |
+
],
|
| 62 |
+
"dtype": "int64"
|
| 63 |
+
}
|
| 64 |
+
],
|
| 65 |
+
"outputs": [
|
| 66 |
+
{
|
| 67 |
+
"shape": [
|
| 68 |
+
1,
|
| 69 |
+
768
|
| 70 |
+
],
|
| 71 |
+
"dtype": "float32"
|
| 72 |
+
}
|
| 73 |
+
]
|
| 74 |
+
}
|
| 75 |
+
}
|
| 76 |
}
|
| 77 |
]
|
| 78 |
}
|
xnnpack/multi_qa_mpnet_base_dot_v1_xnnpack_8da4w.pte
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4942c08d111e825df300d90ae780882d457721ab39f27d731031db97ee3bdf68
|
| 3 |
+
size 72474880
|