trunk/c96c0d945cc30cd4317ca02911049a75e43972b7: flex_gemm: separate layout recognition from reduction analysis (#198421)
- PyTorch: 1668 events in the last 90 days
- PyTorch: 1649th Release in the last 90 days
- Previous: earlier the same day · ciflow/torchtitan/199089
What happened
AI-assisted summary of the author-requested refactor-only base: Move grouped-view recognition and main-output contraction matching into gemm_epilogue_layout.py . Keep reduction dataflow and output ownership in gemm_epilogue_analysis.py ; codegen and the existing layout tests import recognition from its canonical owner. This is a behavior-preserving refactor. Existing recognition bodies, supported expressions, guards…
Summary assembled by rule from the sources below