Hugging Face's logo

Kernels:
webgpu-kernels
/
ai.onnx.ReduceMin

Kernel card Files
xet
 Community
Xenova HF Staff commited on
Commit
b8d19e1
·
verified ·
1 Parent(s): 96d2c8a

sync c928d21e6cc1

Browse files
Files changed (2) hide show
  1. build/webgpu/metadata.json +3 -3
  2. build/webgpu/reduce-row-tree.wgsl.jinja +2 -1
build/webgpu/metadata.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "name": "ai.onnx.ReduceMin",
3
- "id": "_ai_onnx_reducemin_webgpu_a5e3e3d",
4
  "version": 1,
5
  "license": "Apache-2.0",
6
  "backend": { "type": "webgpu" },
@@ -18,11 +18,11 @@
18
  "reduce-narrow-empty-identity.wgsl.jinja": "S6UnLW+6dyCb9ciRezkgXE9mlYmqdfBCjPRO9iPm3Jk=",
19
  "reduce-noop-empty-axes.wgsl.jinja": "IO2MEzyBGJr3Z8AVP7ELJXqfhwcG/vtBfa0wCdkbuwo=",
20
  "reduce-row-subgroup.wgsl.jinja": "szYjtA6TUPJ/I4mbVX7WZUoXiseQ3JfulzqgGb6fdt4=",
21
- "reduce-row-tree.wgsl.jinja": "srFhdHteraJrXxSPG56B56yK9qjUjRAzCEakQKBsuLY=",
22
  "reduce-serial-axis.wgsl.jinja": "cLrTc0ite8YZmx/wn7PBwaxTp9dLWWUIUdQQ3tH8+4o=",
23
  "test.json": "G8Sv1RfEOyXjQTYVWqiWogZbCuoav1pCHqswBcjcM6E="
24
  }
25
  },
26
- "provenance": { "kernel": { "sha": "2e7068faf55e7f43df740015f6d1ee49391a41c5", "dirty": false } },
27
  "webgpu": { "manifestSpec": "1.0", "specialized": true, "opPath": "ops/ai.onnx.ReduceMin" }
28
  }
 
1
  {
2
  "name": "ai.onnx.ReduceMin",
3
+ "id": "_ai_onnx_reducemin_webgpu_dfa003d",
4
  "version": 1,
5
  "license": "Apache-2.0",
6
  "backend": { "type": "webgpu" },
 
18
  "reduce-narrow-empty-identity.wgsl.jinja": "S6UnLW+6dyCb9ciRezkgXE9mlYmqdfBCjPRO9iPm3Jk=",
19
  "reduce-noop-empty-axes.wgsl.jinja": "IO2MEzyBGJr3Z8AVP7ELJXqfhwcG/vtBfa0wCdkbuwo=",
20
  "reduce-row-subgroup.wgsl.jinja": "szYjtA6TUPJ/I4mbVX7WZUoXiseQ3JfulzqgGb6fdt4=",
21
+ "reduce-row-tree.wgsl.jinja": "flhq6QZkgMeHyb1b5enejKpaLEoys2Wnqn9qm1a3Tr8=",
22
  "reduce-serial-axis.wgsl.jinja": "cLrTc0ite8YZmx/wn7PBwaxTp9dLWWUIUdQQ3tH8+4o=",
23
  "test.json": "G8Sv1RfEOyXjQTYVWqiWogZbCuoav1pCHqswBcjcM6E="
24
  }
25
  },
26
+ "provenance": { "kernel": { "sha": "c928d21e6cc1310861cba3bafb75f5f679ecf5f3", "dirty": false } },
27
  "webgpu": { "manifestSpec": "1.0", "specialized": true, "opPath": "ops/ai.onnx.ReduceMin" }
28
  }
build/webgpu/reduce-row-tree.wgsl.jinja CHANGED
@@ -39,7 +39,8 @@ fn {{ name }}() -> {{ scalar }} {
39
  const WG: u32 = {{ workgroupSize }}u;
40
 
41
  {% set is_int = scalar == "i32" or scalar == "u32" %}
42
- {% set intAdditive = is_int and (source.op == "sum" or source.op == "l1" or source.op == "sumsquare") %}
 
43
  {% set accType = scalar if source.op == "max" or source.op == "min" or source.op == "prod" or intAdditive else "f32" %}
44
  {{ wgsl_minmax_identity("identity", source.op, accType) }}
45
 
 
39
  const WG: u32 = {{ workgroupSize }}u;
40
 
41
  {% set is_int = scalar == "i32" or scalar == "u32" %}
42
+ {% set intAdditive = is_int and (source.op == "sum" or source.op == "l1" or source.op == "sumsquare"
43
+ or source.op == "l2" or source.op == "mean") %}
44
  {% set accType = scalar if source.op == "max" or source.op == "min" or source.op == "prod" or intAdditive else "f32" %}
45
  {{ wgsl_minmax_identity("identity", source.op, accType) }}
46