Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[WIP][CPU] support transform-dialect based bf16 gemm codegen #1176

Open
wants to merge 3 commits into
base: main
Choose a base branch
from

Conversation

Guo-Peilin
Copy link
Collaborator

No description provided.

@Guo-Peilin Guo-Peilin changed the title [WIP] support transform-dialect based bf16 gemm codegen [WIP][CPU] support transform-dialect based bf16 gemm codegen Jun 1, 2023
@Guo-Peilin Guo-Peilin force-pushed the support-bf16-gemm-codegen branch from e0df0ff to ede3f27 Compare June 1, 2023 07:35
1. add support for uzp1/uzp2 intrinsics
2. add 2x2x4, 4x4x8, 8x8x8, 8x4x40 bfmmla micro kernels
TODO: input and weight prepacking and prepadding
@Guo-Peilin Guo-Peilin force-pushed the support-bf16-gemm-codegen branch from ede3f27 to b757a2c Compare June 19, 2023 02:13
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