Repository navigation

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

Merged

JCNTH merged 11 commits into

gh/JCNTH/75/origfrom

gh/JCNTH/76/orig

Jul 21, 2026

Commits on Jul 21, 2026

[ExecuTorch][Vulkan] Add et_vk.adamw_step kernel (on-GPU AdamW optimizer step)
JCNTH
committed
c820f6a
[ExecuTorch][Vulkan] Tests for et_vk.adamw_step
JCNTH
committed
1e7e379
[ExecuTorch][Vulkan] Add et_vk.linear_dW kernel (dense fp32 weight-grad GEMM)
JCNTH
committed
9d89ae9
[ExecuTorch][Vulkan] Tests for et_vk.linear_dW
JCNTH
committed
c37b58c
[ExecuTorch][Vulkan] Add et_vk.fused_ce kernels (fused cross-entropy loss + dlogits)
JCNTH
committed
c06a7d3
[ExecuTorch][Vulkan] Tests for et_vk.fused_ce
JCNTH
committed
114c96f
[ExecuTorch][Vulkan] Add et_vk.linear_q4gsw_backward kernel (4-bit input-grad)
JCNTH
committed
179dc9e
[ExecuTorch][Vulkan] Tests for et_vk.linear_q4gsw_backward
JCNTH
committed
eff29fd
[ExecuTorch][Vulkan] Add et_vk.q4gsw_requant kernel (STE re-quant to W_4X8)
JCNTH
committed
c1d1ca1
[ExecuTorch][Vulkan] Tests for et_vk.q4gsw_requant
JCNTH
committed
c7c4299
[ExecuTorch][WebGPU] Glob runtime/ops sources in CMakeLists
JCNTH
committed
bc7a941