llvm-project

d4d28914 - [mlir][vector] Add distribution pattern for vector.create_mask (#71619)

[mlir][vector] Add distribution pattern for vector.create_mask (#71619) This is the last step needed for basic support for distributing masked vector code. The lane id gets delinearized based on the distributed mask shape and then compared against the original mask sizes to compute the bounds for the distributed mask. Note that the distribution of masks is implicit on the shape specified by the warp op. As a result, it is the responsibility of the consumer of the mask to ensure the distributed mask will match its own distribution semantics.