Skip to content

[ARK] Improve XPU W4A16 WOQ decode with dense S4 DPAS path - #2331

Draft
Zhenzhong1 wants to merge 2 commits into
mainfrom
zhenzhong/decode_opt
Draft

[ARK] Improve XPU W4A16 WOQ decode with dense S4 DPAS path#2331
Zhenzhong1 wants to merge 2 commits into
mainfrom
zhenzhong/decode_opt

Conversation

@Zhenzhong1

Copy link
Copy Markdown
Contributor

Description

This PR adds a dense S4 DPAS WOQ path for XPU W4A16 decode when m > 1. The new path avoids the previous generic fallback for small batched decode and directly launches dense DPAS tiles for symmetric INT4 weights with FP16 activations/scales/output.

Summary:

  • Add sycl_tla_dense_woq_s4_dpas.hpp for dense W4A16/S4 DPAS GEMM.
  • Route m > 1 WOQ decode through the dense S4 DPAS path when dtype/shape conditions match.

Signed-off-by: Zhenzhong1 <zhenzhong.xu@intel.com>
Signed-off-by: Zhenzhong1 <zhenzhong.xu@intel.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant