trunk/31a88272c810ba7dd6fec77d5f0fc2e5ca3f7ad4: [ROCm] Added CK-Tile based Group-GEMM backend for ROCm (#200307)
- PyTorch: 2339 events in the last 90 days
- PyTorch: 2309th Release in the last 90 days
- Previous: earlier the same day · trunk/3ffb00d1da9142c97113b50d0ede8e65c34472cf
What happened
Linked issue or supporting maintainer Supported by @jeffdaily This PR adds a minimal CK-Tile based group GEMM backend, that can be enabled via the env var flag ROCM_ALLOW_GROUP_GEMM_CK_TILE . Note that this is the first PR in a series of more PRs that will gradually add more capability, support, performance, etc... The goal of this PR is just to get things to a working state. Currently in this implementation: The on…
Summary assembled by rule from the sources below