maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

[MLIR] NFC. gpu.launch op argument const folder cleanup 

NFC updates to gpu.launch op argument const folder.

Differential Revision: https://reviews.llvm.org/D110136
This commit is contained in:
Uday Bondhugula 2021-09-21 14:23:22 +05:30
parent 7e7484a816
commit 5c77ed0330
1 changed files with 3 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
mlir/lib/Dialect/GPU/IR/GPUDialect.cpp
View File

@ -532,23 +532,19 @@ static ParseResult parseLaunchOp(OpAsmParser &parser, OperationState &result) {
parser.parseOptionalAttrDict(result.attributes));
}
/// Simplify the gpu.launch when the range of the thread and block IDs is
/// Simplify the gpu.launch when the range of a thread or block ID is
/// trivially known to be one.
struct FoldLaunchArguments : public OpRewritePattern<LaunchOp> {
using OpRewritePattern<LaunchOp>::OpRewritePattern;
LogicalResult matchAndRewrite(LaunchOp op,
PatternRewriter &rewriter) const override {
auto isTriviallyOne = [](Value size) {
IntegerAttr cst;
return matchPattern(size, m_Constant(&cst)) && cst.getInt() == 1;
};
// If the range implies a single value for `id`, replace `id`'s uses by
// zero.
Value zero;
bool simplified = false;
auto constPropIdUses = [&](Value id, Value size) {
if (!isTriviallyOne(size))
// Check if size is trivially one.
if (!matchPattern(size, m_One()))
return;
if (!simplified) {
// Create a zero value the first time.