Unverified Commit d4d28914 authored by Quinn Dawkins's avatar Quinn Dawkins Committed by GitHub
Browse files

[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.
parent 22c68511
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment