[ExecuTorch][WebGPU] Add softmax ops (_softmax, _log_softmax) to the WebGPU backend by pytorchbot · Pull Request #21072 · pytorch/executorch

[ExecuTorch][WebGPU] Add softmax ops (_softmax, _log_softmax) to the WebGPU backend

JCNTH

committed

[ExecuTorch][WebGPU] Tests for the softmax WebGPU ops

JCNTH

committed

[ExecuTorch][WebGPU] Add elementwise binary ops (div, sub) to the WebGPU backend

JCNTH

committed

[ExecuTorch][WebGPU] Tests for the binary WebGPU ops

JCNTH

committed

[ExecuTorch][WebGPU] Add reduce op (sum/mean) to the WebGPU backend

JCNTH

committed

[ExecuTorch][WebGPU] Tests for the reduce WebGPU ops

JCNTH

committed

[ExecuTorch][WebGPU] Add select/boolean ops (where, scalar compares, logical_not) to the WebGPU backend

JCNTH

committed

[ExecuTorch][WebGPU] Tests for the select_bool WebGPU ops

JCNTH

committed

[ExecuTorch][WebGPU] Add gather and embedding ops to the WebGPU backend

JCNTH

committed

[ExecuTorch][WebGPU] Tests for the gather WebGPU ops

JCNTH

committed

[ExecuTorch][WebGPU] Add shape/copy ops (dim_order, expand_copy, fill) to the WebGPU backend

JCNTH

committed

[ExecuTorch][WebGPU] Tests for the shape_copy WebGPU ops

JCNTH

committed

[ExecuTorch][WebGPU] Add q4gsw training kernels (backward, weight-grad, requant) to the WebGPU backend

JCNTH

committed

[ExecuTorch][WebGPU] Tests for the q4gsw_train WebGPU ops

JCNTH

committed

[ExecuTorch][WebGPU] Add et_vk.fused_ce (fused cross-entropy) to the WebGPU backend

JCNTH

committed

[ExecuTorch][WebGPU] Tests for the fused_ce WebGPU ops

JCNTH

committed

[ExecuTorch][WebGPU] Add et_vk.adamw_step (on-GPU AdamW optimizer step) to the WebGPU backend

JCNTH

committed

[ExecuTorch][WebGPU] Tests for the adamw WebGPU ops

JCNTH

committed

[ExecuTorch][Vulkan] Add et_vk.adamw_step kernel (on-GPU AdamW optimizer step)

JCNTH

committed

[ExecuTorch][Vulkan] Tests for et_vk.adamw_step

JCNTH

committed

[ExecuTorch][Vulkan] Add et_vk.linear_dW kernel (dense fp32 weight-grad GEMM)

JCNTH

committed

[ExecuTorch][Vulkan] Tests for et_vk.linear_dW

JCNTH

committed

[ExecuTorch][Vulkan] Add et_vk.fused_ce kernels (fused cross-entropy loss + dlogits)

JCNTH

committed

[ExecuTorch][Vulkan] Tests for et_vk.fused_ce

JCNTH

committed

[ExecuTorch][Vulkan] Add et_vk.linear_q4gsw_backward kernel (4-bit input-grad)

JCNTH

committed

[ExecuTorch][Vulkan] Tests for et_vk.linear_q4gsw_backward

JCNTH

committed

[ExecuTorch][Vulkan] Add et_vk.q4gsw_requant kernel (STE re-quant to W_4X8)

JCNTH

committed

[ExecuTorch][Vulkan] Tests for et_vk.q4gsw_requant

JCNTH

committed

[ExecuTorch][WebGPU] Glob runtime/ops sources in CMakeLists

JCNTH

committed

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

[ExecuTorch][WebGPU] Add softmax ops (_softmax, _log_softmax) to the WebGPU backend#21072

[ExecuTorch][WebGPU] Add softmax ops (_softmax, _log_softmax) to the WebGPU backend#21072
JCNTH merged 29 commits into
gh/JCNTH/57/origfrom
gh/JCNTH/58/orig

Commits on Jul 21, 2026

[ExecuTorch][WebGPU] Add softmax ops (_softmax, _log_softmax) to the WebGPU backend

[ExecuTorch][WebGPU] Tests for the softmax WebGPU ops

[ExecuTorch][WebGPU] Add elementwise binary ops (div, sub) to the WebGPU backend

[ExecuTorch][WebGPU] Tests for the binary WebGPU ops

[ExecuTorch][WebGPU] Add reduce op (sum/mean) to the WebGPU backend

[ExecuTorch][WebGPU] Tests for the reduce WebGPU ops

[ExecuTorch][WebGPU] Add select/boolean ops (where, scalar compares, logical_not) to the WebGPU backend

[ExecuTorch][WebGPU] Tests for the select_bool WebGPU ops

[ExecuTorch][WebGPU] Add gather and embedding ops to the WebGPU backend

[ExecuTorch][WebGPU] Tests for the gather WebGPU ops

[ExecuTorch][WebGPU] Add shape/copy ops (dim_order, expand_copy, fill) to the WebGPU backend

[ExecuTorch][WebGPU] Tests for the shape_copy WebGPU ops

[ExecuTorch][WebGPU] Add q4gsw training kernels (backward, weight-grad, requant) to the WebGPU backend

[ExecuTorch][WebGPU] Tests for the q4gsw_train WebGPU ops

[ExecuTorch][WebGPU] Add et_vk.fused_ce (fused cross-entropy) to the WebGPU backend

[ExecuTorch][WebGPU] Tests for the fused_ce WebGPU ops

[ExecuTorch][WebGPU] Add et_vk.adamw_step (on-GPU AdamW optimizer step) to the WebGPU backend

[ExecuTorch][WebGPU] Tests for the adamw WebGPU ops

[ExecuTorch][Vulkan] Add et_vk.adamw_step kernel (on-GPU AdamW optimizer step)

[ExecuTorch][Vulkan] Tests for et_vk.adamw_step

[ExecuTorch][Vulkan] Add et_vk.linear_dW kernel (dense fp32 weight-grad GEMM)

[ExecuTorch][Vulkan] Tests for et_vk.linear_dW

[ExecuTorch][Vulkan] Add et_vk.fused_ce kernels (fused cross-entropy loss + dlogits)

[ExecuTorch][Vulkan] Tests for et_vk.fused_ce

[ExecuTorch][Vulkan] Add et_vk.linear_q4gsw_backward kernel (4-bit input-grad)

[ExecuTorch][Vulkan] Tests for et_vk.linear_q4gsw_backward

[ExecuTorch][Vulkan] Add et_vk.q4gsw_requant kernel (STE re-quant to W_4X8)

[ExecuTorch][Vulkan] Tests for et_vk.q4gsw_requant

[ExecuTorch][WebGPU] Glob runtime/ops sources in CMakeLists