HomePhabricator

[mlir][gpu] Relax restriction on MMA store op to allow chain of mma ops.

Authored by ThomasRaoux on May 27 2021, 8:58 AM.

Description

[mlir][gpu] Relax restriction on MMA store op to allow chain of mma ops.

In order to allow large matmul operations using the MMA ops we need to chain
operations this is not possible unless "DOp" and "COp" type have matching
layout so remove the "DOp" layout and force accumulator and result type to
match.
Added a test for the case where the MMA value is accumulated.

Differential Revision: https://reviews.llvm.org/D103023