Sergei Grechanik d80b04ab00 [mlir][Affine][Vector] Support vectorizing reduction loops ... This patch adds support for vectorizing loops with 'iter_args' implementing known reductions along the vector dimension. Comparing to the non-vector-dimension case, two additional things are done during vectorization of such loops: - The resulting vector returned from the loop is reduced to a scalar using `vector.reduce`. - In some cases a mask is applied to the vector yielded at the end of the loop to prevent garbage values from being written to the accumulator. Vectorization of reduction loops is disabled by default. To enable it, a map from loops to array of reduction descriptors should be explicitly passed to `vectorizeAffineLoops`, or `vectorize-reductions=true` should be passed to the SuperVectorize pass. Current limitations: - Loops with a non-unit step size are not supported. - n-D vectorization with n > 1 is not supported. Reviewed By: nicolasvasilache Differential Revision: https://reviews.llvm.org/D100694		2021-05-05 09:03:59 -07:00
..
AffineToStandard.cpp	[mlir][Affine][Vector] Support vectorizing reduction loops	2021-05-05 09:03:59 -07:00
CMakeLists.txt	[MLIR] Create memref dialect and move dialect-specific ops from std.	2021-03-15 11:14:09 +01:00