[Mlir-commits] [mlir] [mlir][emitc] Mark emitc.literal with CExpression (PR #155582)
Vlad Lazar
llvmlistbot at llvm.org
Wed Aug 27 07:14:29 PDT 2025
https://github.com/Vladislave0-0 updated https://github.com/llvm/llvm-project/pull/155582
>From d2c852c55d7d8bf79249302b4fc9222341624988 Mon Sep 17 00:00:00 2001
From: Vlad Lazar <lazar_2004 at list.ru>
Date: Wed, 27 Aug 2025 12:25:15 +0300
Subject: [PATCH 1/2] [mlir][emitc] Mark emitc.literal with CExpression
This PR marks `emitc.literal` as a valid operation within
`emitc.expression`, removing an artificial restriction since
literals are inherently inlined into expressions during emission.
Example:
```mlir
%0 = emitc.expression : i32 {
%literal = emitc.literal "42" : i32
emitc.yield %literal : i32
}
```
---
mlir/include/mlir/Dialect/EmitC/IR/EmitC.td | 2 +-
mlir/lib/Target/Cpp/TranslateToCpp.cpp | 23 ++++++++++------
mlir/test/Dialect/EmitC/form-expressions.mlir | 26 +++++++++++++++++++
mlir/test/Target/Cpp/expressions.mlir | 26 +++++++++++++++++++
4 files changed, 68 insertions(+), 9 deletions(-)
diff --git a/mlir/include/mlir/Dialect/EmitC/IR/EmitC.td b/mlir/include/mlir/Dialect/EmitC/IR/EmitC.td
index 59beac7d64154..6fbf52b16db31 100644
--- a/mlir/include/mlir/Dialect/EmitC/IR/EmitC.td
+++ b/mlir/include/mlir/Dialect/EmitC/IR/EmitC.td
@@ -851,7 +851,7 @@ def EmitC_IncludeOp
let hasCustomAssemblyFormat = 1;
}
-def EmitC_LiteralOp : EmitC_Op<"literal", [Pure]> {
+def EmitC_LiteralOp : EmitC_Op<"literal", [Pure, CExpressionInterface]> {
let summary = "Literal operation";
let description = [{
The `emitc.literal` operation produces an SSA value equal to some constant
diff --git a/mlir/lib/Target/Cpp/TranslateToCpp.cpp b/mlir/lib/Target/Cpp/TranslateToCpp.cpp
index 7284e24776175..86fa7b0eac4c0 100644
--- a/mlir/lib/Target/Cpp/TranslateToCpp.cpp
+++ b/mlir/lib/Target/Cpp/TranslateToCpp.cpp
@@ -98,6 +98,7 @@ static FailureOr<int> getOperatorPrecedence(Operation *operation) {
.Case<emitc::ConditionalOp>([&](auto op) { return 2; })
.Case<emitc::DivOp>([&](auto op) { return 13; })
.Case<emitc::LoadOp>([&](auto op) { return 16; })
+ .Case<emitc::LiteralOp>([&](auto op) { return 16; })
.Case<emitc::LogicalAndOp>([&](auto op) { return 4; })
.Case<emitc::LogicalNotOp>([&](auto op) { return 15; })
.Case<emitc::LogicalOrOp>([&](auto op) { return 3; })
@@ -440,6 +441,16 @@ static LogicalResult printOperation(CppEmitter &emitter, emitc::LoadOp loadOp) {
return emitter.emitOperand(loadOp.getOperand());
}
+static LogicalResult printOperation(CppEmitter &emitter,
+ emitc::LiteralOp literalOp) {
+ // If literalOp is used inside an expression, we treat it as an embedded one.
+ if (emitter.isPartOfCurrentExpression(literalOp.getResult()))
+ return emitter.ostream() << literalOp.getValue(), success();
+
+ emitter.cacheDeferredOpResult(literalOp.getResult(), literalOp.getValue());
+ return success();
+}
+
static LogicalResult printBinaryOperation(CppEmitter &emitter,
Operation *operation,
StringRef binaryOperator) {
@@ -1688,10 +1699,10 @@ LogicalResult CppEmitter::emitOperation(Operation &op, bool trailingSemicolon) {
emitc::DeclareFuncOp, emitc::DivOp, emitc::ExpressionOp,
emitc::FieldOp, emitc::FileOp, emitc::ForOp, emitc::FuncOp,
emitc::GlobalOp, emitc::IfOp, emitc::IncludeOp, emitc::LoadOp,
- emitc::LogicalAndOp, emitc::LogicalNotOp, emitc::LogicalOrOp,
- emitc::MulOp, emitc::RemOp, emitc::ReturnOp, emitc::SubOp,
- emitc::SwitchOp, emitc::UnaryMinusOp, emitc::UnaryPlusOp,
- emitc::VariableOp, emitc::VerbatimOp>(
+ emitc::LiteralOp, emitc::LogicalAndOp, emitc::LogicalNotOp,
+ emitc::LogicalOrOp, emitc::MulOp, emitc::RemOp, emitc::ReturnOp,
+ emitc::SubOp, emitc::SwitchOp, emitc::UnaryMinusOp,
+ emitc::UnaryPlusOp, emitc::VariableOp, emitc::VerbatimOp>(
[&](auto op) { return printOperation(*this, op); })
// Func ops.
@@ -1705,10 +1716,6 @@ LogicalResult CppEmitter::emitOperation(Operation &op, bool trailingSemicolon) {
cacheDeferredOpResult(op.getResult(), op.getFieldName());
return success();
})
- .Case<emitc::LiteralOp>([&](auto op) {
- cacheDeferredOpResult(op.getResult(), op.getValue());
- return success();
- })
.Case<emitc::MemberOp>([&](auto op) {
cacheDeferredOpResult(op.getResult(), createMemberAccess(op));
return success();
diff --git a/mlir/test/Dialect/EmitC/form-expressions.mlir b/mlir/test/Dialect/EmitC/form-expressions.mlir
index 1ac053bf1215f..e4e8fc0ebbc00 100644
--- a/mlir/test/Dialect/EmitC/form-expressions.mlir
+++ b/mlir/test/Dialect/EmitC/form-expressions.mlir
@@ -160,3 +160,29 @@ func.func @expression_with_load(%arg0: i32, %arg1: !emitc.ptr<i32>) -> i1 {
%c = emitc.cmp lt, %b, %arg0 :(i32, i32) -> i1
return %c : i1
}
+
+// CHECK-LABEL: func.func @expression_with_literal(
+// CHECK-SAME: %[[ARG0:.*]]: i32) -> i1 {
+// CHECK: %[[VAL_0:.*]] = emitc.expression : i32 {
+// CHECK: %[[VAL_1:.*]] = literal "1" : i32
+// CHECK: yield %[[VAL_1]] : i32
+// CHECK: }
+// CHECK: %[[VAL_2:.*]] = emitc.expression : i32 {
+// CHECK: %[[VAL_3:.*]] = literal "2" : i32
+// CHECK: yield %[[VAL_3]] : i32
+// CHECK: }
+// CHECK: %[[VAL_4:.*]] = emitc.expression : i1 {
+// CHECK: %[[VAL_5:.*]] = add %[[VAL_0]], %[[VAL_2]] : (i32, i32) -> i32
+// CHECK: %[[VAL_6:.*]] = cmp lt, %[[VAL_5]], %[[ARG0]] : (i32, i32) -> i1
+// CHECK: yield %[[VAL_6]] : i1
+// CHECK: }
+// CHECK: return %[[VAL_4]] : i1
+// CHECK: }
+
+func.func @expression_with_literal(%arg0: i32) -> i1 {
+ %literal1 = emitc.literal "1" : i32
+ %literal2 = emitc.literal "2" : i32
+ %b = emitc.add %literal1, %literal2 : (i32, i32) -> i32
+ %c = emitc.cmp lt, %b, %arg0 :(i32, i32) -> i1
+ return %c : i1
+}
diff --git a/mlir/test/Target/Cpp/expressions.mlir b/mlir/test/Target/Cpp/expressions.mlir
index 5057d6ca0c4cf..b5cea8e5e9bbf 100644
--- a/mlir/test/Target/Cpp/expressions.mlir
+++ b/mlir/test/Target/Cpp/expressions.mlir
@@ -439,3 +439,29 @@ emitc.func @expression_with_call_opaque_with_args_array(%0 : i32, %1 : i32) {
}
return
}
+
+
+// CPP-DEFAULT: bool expression_with_literal(int32_t [[VAL_1:v[0-9]+]]) {
+// CPP-DEFAULT-NEXT: bool [[VAL_2:v[0-9]+]] = (1 + [[VAL_1]]) / 2 < 3;
+// CPP-DEFAULT-NEXT: return [[VAL_2]];
+// CPP-DEFAULT-NEXT: }
+
+// CPP-DECLTOP: bool expression_with_literal(int32_t [[VAL_1:v[0-9]+]]) {
+// CPP-DECLTOP-NEXT: bool [[VAL_2:v[0-9]+]];
+// CPP-DECLTOP-NEXT: [[VAL_2]] = (1 + [[VAL_1]]) / 2 < 3;
+// CPP-DECLTOP-NEXT: return [[VAL_2]];
+// CPP-DECLTOP-NEXT: }
+
+func.func @expression_with_literal(%arg0 : i32) -> i1 {
+ %ret = emitc.expression noinline : i1 {
+ %literal1 = emitc.literal "1" : i32
+ %literal2 = emitc.literal "2" : i32
+ %add = add %literal1, %arg0 : (i32, i32) -> i32
+ %div = div %add, %literal2 : (i32, i32) -> i32
+ %literal3 = emitc.literal "3" : i32
+ %f = emitc.cmp lt, %div, %literal3 :(i32, i32) -> i1
+ emitc.yield %f : i1
+ }
+
+ return %ret : i1
+}
>From bbe2aa9d98443bea34b8ab14e0f0c038b6a25098 Mon Sep 17 00:00:00 2001
From: Vlad Lazar <lazar_2004 at list.ru>
Date: Wed, 27 Aug 2025 17:12:45 +0300
Subject: [PATCH 2/2] [mlir][emitc] Changes after review
---
mlir/include/mlir/Dialect/EmitC/IR/EmitC.td | 6 ++++++
mlir/lib/Target/Cpp/TranslateToCpp.cpp | 2 +-
mlir/test/Dialect/EmitC/form-expressions.mlir | 18 ++++++------------
mlir/test/Target/Cpp/expressions.mlir | 17 +++++++++++++++++
4 files changed, 30 insertions(+), 13 deletions(-)
diff --git a/mlir/include/mlir/Dialect/EmitC/IR/EmitC.td b/mlir/include/mlir/Dialect/EmitC/IR/EmitC.td
index 6fbf52b16db31..18595f5786280 100644
--- a/mlir/include/mlir/Dialect/EmitC/IR/EmitC.td
+++ b/mlir/include/mlir/Dialect/EmitC/IR/EmitC.td
@@ -874,6 +874,12 @@ def EmitC_LiteralOp : EmitC_Op<"literal", [Pure, CExpressionInterface]> {
let hasVerifier = 1;
let assemblyFormat = "$value attr-dict `:` type($result)";
+
+ let extraClassDeclaration = [{
+ bool hasSideEffects() {
+ return false;
+ }
+ }];
}
def EmitC_LogicalAndOp : EmitC_BinaryOp<"logical_and", []> {
diff --git a/mlir/lib/Target/Cpp/TranslateToCpp.cpp b/mlir/lib/Target/Cpp/TranslateToCpp.cpp
index 86fa7b0eac4c0..61f9575c3150a 100644
--- a/mlir/lib/Target/Cpp/TranslateToCpp.cpp
+++ b/mlir/lib/Target/Cpp/TranslateToCpp.cpp
@@ -98,7 +98,7 @@ static FailureOr<int> getOperatorPrecedence(Operation *operation) {
.Case<emitc::ConditionalOp>([&](auto op) { return 2; })
.Case<emitc::DivOp>([&](auto op) { return 13; })
.Case<emitc::LoadOp>([&](auto op) { return 16; })
- .Case<emitc::LiteralOp>([&](auto op) { return 16; })
+ .Case<emitc::LiteralOp>([&](auto op) { return 17; })
.Case<emitc::LogicalAndOp>([&](auto op) { return 4; })
.Case<emitc::LogicalNotOp>([&](auto op) { return 15; })
.Case<emitc::LogicalOrOp>([&](auto op) { return 3; })
diff --git a/mlir/test/Dialect/EmitC/form-expressions.mlir b/mlir/test/Dialect/EmitC/form-expressions.mlir
index e4e8fc0ebbc00..015133489d566 100644
--- a/mlir/test/Dialect/EmitC/form-expressions.mlir
+++ b/mlir/test/Dialect/EmitC/form-expressions.mlir
@@ -163,20 +163,14 @@ func.func @expression_with_load(%arg0: i32, %arg1: !emitc.ptr<i32>) -> i1 {
// CHECK-LABEL: func.func @expression_with_literal(
// CHECK-SAME: %[[ARG0:.*]]: i32) -> i1 {
-// CHECK: %[[VAL_0:.*]] = emitc.expression : i32 {
+// CHECK: %[[VAL_0:.*]] = emitc.expression : i1 {
// CHECK: %[[VAL_1:.*]] = literal "1" : i32
-// CHECK: yield %[[VAL_1]] : i32
+// CHECK: %[[VAL_2:.*]] = literal "2" : i32
+// CHECK: %[[VAL_3:.*]] = add %[[VAL_1]], %[[VAL_2]] : (i32, i32) -> i32
+// CHECK: %[[VAL_4:.*]] = cmp lt, %[[VAL_3]], %[[ARG0]] : (i32, i32) -> i1
+// CHECK: yield %[[VAL_4]] : i1
// CHECK: }
-// CHECK: %[[VAL_2:.*]] = emitc.expression : i32 {
-// CHECK: %[[VAL_3:.*]] = literal "2" : i32
-// CHECK: yield %[[VAL_3]] : i32
-// CHECK: }
-// CHECK: %[[VAL_4:.*]] = emitc.expression : i1 {
-// CHECK: %[[VAL_5:.*]] = add %[[VAL_0]], %[[VAL_2]] : (i32, i32) -> i32
-// CHECK: %[[VAL_6:.*]] = cmp lt, %[[VAL_5]], %[[ARG0]] : (i32, i32) -> i1
-// CHECK: yield %[[VAL_6]] : i1
-// CHECK: }
-// CHECK: return %[[VAL_4]] : i1
+// CHECK: return %[[VAL_0]] : i1
// CHECK: }
func.func @expression_with_literal(%arg0: i32) -> i1 {
diff --git a/mlir/test/Target/Cpp/expressions.mlir b/mlir/test/Target/Cpp/expressions.mlir
index b5cea8e5e9bbf..bd279c160d512 100644
--- a/mlir/test/Target/Cpp/expressions.mlir
+++ b/mlir/test/Target/Cpp/expressions.mlir
@@ -465,3 +465,20 @@ func.func @expression_with_literal(%arg0 : i32) -> i1 {
return %ret : i1
}
+
+
+// CPP-DEFAULT: int32_t single_literal_in_expression() {
+// CPP-DEFAULT-NEXT: return 42;
+// CPP-DEFAULT-NEXT: }
+
+// CPP-DECLTOP: int32_t single_literal_in_expression() {
+// CPP-DECLTOP-NEXT: return 42;
+// CPP-DECLTOP-NEXT: }
+
+func.func @single_literal_in_expression() -> i32 {
+ %result = emitc.expression : i32 {
+ %lit = emitc.literal "42" : i32
+ emitc.yield %lit : i32
+ }
+ return %result : i32
+}
More information about the Mlir-commits
mailing list