trunk/640f90e15bc0e9af5caaa054939015d14a086131: flex_gemm: compose paired fragment reductions (#192894)
- PyTorch 近 90 天出现 2303 次
- PyTorch 近 90 天第 2273 次发版
- 上一次:同一天稍早 · trunk/10dc63912abee5b1c6b942e670872e515cbf9210: [CPP FakeTensor] Don't look up the op in register_fake (#200798)
发生了什么
Human Note At a high level we are now allowing chunking + reductions along the n axis. So this is two forms of reductions; 1 is the chunked or interleave pairing and the other is the reduction along the N axis follwing this up. Below we have 2 for the first and then 32 for the second or a total of 64 elements. We needed so mechanize to enforce we have 64 acc values in regsisters and we do the reductions only in regi…
摘要按规则整理自下方来源原文