Hugging Face's logo

Kernels:
webgpu-kernels
/
ai.onnx.OneHot

Kernel card Files
xet
 Community
Xenova HF Staff commited on
Commit
bbc2d6e
·
verified ·
1 Parent(s): 239770a

sync c928d21e6cc1

Browse files
Files changed (2) hide show
  1. build/webgpu/bench.json +7 -14
  2. build/webgpu/metadata.json +3 -3
build/webgpu/bench.json CHANGED
@@ -4,7 +4,7 @@
4
  {
5
  "name": "tokens_vocab",
6
  "inputs": {
7
- "indices": { "dtype": "uint32", "shape": [128] },
8
  "depth": { "dtype": "uint32", "shape": [1], "data": { "kind": "values", "values": [4096] } },
9
  "values": { "dtype": "float32", "shape": [2], "data": { "kind": "values", "values": [0.0, 1.0] } }
10
  },
@@ -14,7 +14,7 @@
14
  "name": "tokens_vocab_2047x8192",
15
  "preset": "smoke",
16
  "inputs": {
17
- "indices": { "dtype": "uint32", "shape": [2047], "dist": "uniform", "seed": 901, "scale": 8000, "offset": 0 },
18
  "depth": { "dtype": "uint32", "shape": [1], "data": { "kind": "values", "values": [8192] } },
19
  "values": { "dtype": "float32", "shape": [2], "data": { "kind": "values", "values": [0.0, 1.0] } }
20
  },
@@ -25,7 +25,7 @@
25
  "name": "unaligned_depth_8191_generic_scalar",
26
  "preset": "smoke",
27
  "inputs": {
28
- "indices": { "dtype": "uint32", "shape": [2047], "dist": "uniform", "seed": 902, "scale": 8000, "offset": 0 },
29
  "depth": { "dtype": "uint32", "shape": [1], "data": { "kind": "values", "values": [8191] } },
30
  "values": { "dtype": "float32", "shape": [2], "data": { "kind": "values", "values": [0.0, 1.0] } }
31
  },
@@ -37,7 +37,7 @@
37
  "preset": "smoke",
38
  "attrs": { "axis": 0 },
39
  "inputs": {
40
- "indices": { "dtype": "uint32", "shape": [2047], "dist": "uniform", "seed": 903, "scale": 8000, "offset": 0 },
41
  "depth": { "dtype": "uint32", "shape": [1], "data": { "kind": "values", "values": [8192] } },
42
  "values": { "dtype": "float32", "shape": [2], "data": { "kind": "values", "values": [0.0, 1.0] } }
43
  },
@@ -49,7 +49,7 @@
49
  "preset": "smoke",
50
  "attrs": { "axis": 1 },
51
  "inputs": {
52
- "indices": { "dtype": "uint32", "shape": [2047, 8], "dist": "uniform", "seed": 904, "scale": 1000, "offset": 0 },
53
  "depth": { "dtype": "uint32", "shape": [1], "data": { "kind": "values", "values": [1024] } },
54
  "values": { "dtype": "float32", "shape": [2], "data": { "kind": "values", "values": [0.0, 1.0] } }
55
  },
@@ -60,7 +60,7 @@
60
  "name": "tokens_vocab_f16_2047x8192_vec4",
61
  "preset": "smoke",
62
  "inputs": {
63
- "indices": { "dtype": "uint32", "shape": [2047], "dist": "uniform", "seed": 905, "scale": 8000, "offset": 0 },
64
  "depth": { "dtype": "uint32", "shape": [1], "data": { "kind": "values", "values": [8192] } },
65
  "values": { "dtype": "float16", "shape": [2], "data": { "kind": "values", "values": [0.0, 1.0] } }
66
  },
@@ -72,14 +72,7 @@
72
  "preset": "stress",
73
  "attrs": { "axis": 1 },
74
  "inputs": {
75
- "indices": {
76
- "dtype": "uint32",
77
- "shape": [256, 8, 8, 8],
78
- "dist": "uniform",
79
- "seed": 911,
80
- "scale": 29,
81
- "offset": 0
82
- },
83
  "depth": { "dtype": "uint32", "shape": [1], "data": { "kind": "values", "values": [30] } },
84
  "values": { "dtype": "float32", "shape": [2], "data": { "kind": "values", "values": [0.0, 1.0] } }
85
  },
 
4
  {
5
  "name": "tokens_vocab",
6
  "inputs": {
7
+ "indices": { "dtype": "uint32", "shape": [128], "min": 0, "max": 4096 },
8
  "depth": { "dtype": "uint32", "shape": [1], "data": { "kind": "values", "values": [4096] } },
9
  "values": { "dtype": "float32", "shape": [2], "data": { "kind": "values", "values": [0.0, 1.0] } }
10
  },
 
14
  "name": "tokens_vocab_2047x8192",
15
  "preset": "smoke",
16
  "inputs": {
17
+ "indices": { "dtype": "uint32", "shape": [2047], "dist": "uniform", "seed": 901, "min": 0, "max": 8000 },
18
  "depth": { "dtype": "uint32", "shape": [1], "data": { "kind": "values", "values": [8192] } },
19
  "values": { "dtype": "float32", "shape": [2], "data": { "kind": "values", "values": [0.0, 1.0] } }
20
  },
 
25
  "name": "unaligned_depth_8191_generic_scalar",
26
  "preset": "smoke",
27
  "inputs": {
28
+ "indices": { "dtype": "uint32", "shape": [2047], "dist": "uniform", "seed": 902, "min": 0, "max": 8000 },
29
  "depth": { "dtype": "uint32", "shape": [1], "data": { "kind": "values", "values": [8191] } },
30
  "values": { "dtype": "float32", "shape": [2], "data": { "kind": "values", "values": [0.0, 1.0] } }
31
  },
 
37
  "preset": "smoke",
38
  "attrs": { "axis": 0 },
39
  "inputs": {
40
+ "indices": { "dtype": "uint32", "shape": [2047], "dist": "uniform", "seed": 903, "min": 0, "max": 8000 },
41
  "depth": { "dtype": "uint32", "shape": [1], "data": { "kind": "values", "values": [8192] } },
42
  "values": { "dtype": "float32", "shape": [2], "data": { "kind": "values", "values": [0.0, 1.0] } }
43
  },
 
49
  "preset": "smoke",
50
  "attrs": { "axis": 1 },
51
  "inputs": {
52
+ "indices": { "dtype": "uint32", "shape": [2047, 8], "dist": "uniform", "seed": 904, "min": 0, "max": 1000 },
53
  "depth": { "dtype": "uint32", "shape": [1], "data": { "kind": "values", "values": [1024] } },
54
  "values": { "dtype": "float32", "shape": [2], "data": { "kind": "values", "values": [0.0, 1.0] } }
55
  },
 
60
  "name": "tokens_vocab_f16_2047x8192_vec4",
61
  "preset": "smoke",
62
  "inputs": {
63
+ "indices": { "dtype": "uint32", "shape": [2047], "dist": "uniform", "seed": 905, "min": 0, "max": 8000 },
64
  "depth": { "dtype": "uint32", "shape": [1], "data": { "kind": "values", "values": [8192] } },
65
  "values": { "dtype": "float16", "shape": [2], "data": { "kind": "values", "values": [0.0, 1.0] } }
66
  },
 
72
  "preset": "stress",
73
  "attrs": { "axis": 1 },
74
  "inputs": {
75
+ "indices": { "dtype": "uint32", "shape": [256, 8, 8, 8], "dist": "uniform", "seed": 911, "min": 0, "max": 29 },
 
 
 
 
 
 
 
76
  "depth": { "dtype": "uint32", "shape": [1], "data": { "kind": "values", "values": [30] } },
77
  "values": { "dtype": "float32", "shape": [2], "data": { "kind": "values", "values": [0.0, 1.0] } }
78
  },
build/webgpu/metadata.json CHANGED
@@ -1,13 +1,13 @@
1
  {
2
  "name": "ai.onnx.OneHot",
3
- "id": "_ai_onnx_onehot_webgpu_21894c9",
4
  "version": 1,
5
  "license": "Apache-2.0",
6
  "backend": { "type": "webgpu" },
7
  "digest": {
8
  "algorithm": "sha256",
9
  "files": {
10
- "bench.json": "8JSemsHPcNMzWPXe0x90can66FWOW5qWvz0P7W7I7Pg=",
11
  "manifest.json": "BfEMlX6Fvw/2pFVCJ1L9FoZ2lBpbHjLoAr2PbX2GBZc=",
12
  "one-hot-fill.wgsl.jinja": "+1+3T4yVF91Ta11a5kEZHAC8D3C/uGzVKZm/CSK4Htc=",
13
  "one-hot-last-axis-vec4.wgsl.jinja": "nQ+vRZPx/rWi3Q+vuIhccDJzW6BATwWLFiaDf+HNwXo=",
@@ -15,6 +15,6 @@
15
  "test.json": "UrZtlg9pSruj4lfpgeNOargkGctAr+ZUIM0fkeBJNsg="
16
  }
17
  },
18
- "provenance": { "kernel": { "sha": "2e7068faf55e7f43df740015f6d1ee49391a41c5", "dirty": false } },
19
  "webgpu": { "manifestSpec": "1.0", "specialized": true, "opPath": "ops/ai.onnx.OneHot" }
20
  }
 
1
  {
2
  "name": "ai.onnx.OneHot",
3
+ "id": "_ai_onnx_onehot_webgpu_439ff74",
4
  "version": 1,
5
  "license": "Apache-2.0",
6
  "backend": { "type": "webgpu" },
7
  "digest": {
8
  "algorithm": "sha256",
9
  "files": {
10
+ "bench.json": "tIK5K4R1/SOMH56dBhsH6mCf7IQ0DE5tsv4L5Q2UuEk=",
11
  "manifest.json": "BfEMlX6Fvw/2pFVCJ1L9FoZ2lBpbHjLoAr2PbX2GBZc=",
12
  "one-hot-fill.wgsl.jinja": "+1+3T4yVF91Ta11a5kEZHAC8D3C/uGzVKZm/CSK4Htc=",
13
  "one-hot-last-axis-vec4.wgsl.jinja": "nQ+vRZPx/rWi3Q+vuIhccDJzW6BATwWLFiaDf+HNwXo=",
 
15
  "test.json": "UrZtlg9pSruj4lfpgeNOargkGctAr+ZUIM0fkeBJNsg="
16
  }
17
  },
18
+ "provenance": { "kernel": { "sha": "c928d21e6cc1310861cba3bafb75f5f679ecf5f3", "dirty": false } },
19
  "webgpu": { "manifestSpec": "1.0", "specialized": true, "opPath": "ops/ai.onnx.OneHot" }
20
  }