trunk/31a88272c810ba7dd6fec77d5f0fc2e5ca3f7ad4: [ROCm] Added CK-Tile based Group-GEMM backend for ROCm (#200307)
- PyTorch 近 90 天出现 2342 次
- PyTorch 近 90 天第 2312 次发版
- 上一次：同一天稍早 · trunk/3ffb00d1da9142c97113b50d0ede8e65c34472cf
发生了什么
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…
摘要按规则整理自下方来源原文