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

[CK Tile] Need GroupGEMM with make_argument on device #1890

Open
zjing14 opened this issue Feb 14, 2025 · 1 comment
Open

[CK Tile] Need GroupGEMM with make_argument on device #1890

zjing14 opened this issue Feb 14, 2025 · 1 comment
Assignees

Comments

@zjing14
Copy link
Contributor

zjing14 commented Feb 14, 2025

Currently, CK Tile GroupGEMM prepares metadata on the host, which requires transferring meta data between the device and host back and forth.

CK_TILE_HOST static auto MakeKargs(const std::vector<GroupedGemmHostArgs>& gemm_descs)

To avoid this overhead, we need groupGEMM kernel as old CK, which is a persistent kernel that reads GEMM shapes from device memory, and calculate offset and block_id on-the-fly.

@zjing14
Copy link
Contributor Author

zjing14 commented Feb 14, 2025

@aosewski Could you help? Move your groupGEMM design from old CK to CK Tile

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

3 participants