viable/strict/1792232715: flex_gemm: support chunked GLU epilogue outputs (#201199)
- PyTorch: 2458 events in the last 90 days
- PyTorch: 2425th Release in the last 90 days
- Previous: earlier the same day · viable/strict/1792229213
What happened
Summary (human written only) Allow for the chunked form of interleaved weights Agent details (AI-generated) Supports concatenated gate/up gradients and saving the unchanged accumulator from a chunked GLU forward epilogue. Existing tile stores write into aligned halves of one allocation; no new CUDA kernel or vendored QuACK changes are needed. Scalar/dtype suffixes and the existing interleaved path are preserved. Tes…
Summary assembled by rule from the sources below