https://github.com/andykaylor updated
https://github.com/llvm/llvm-project/pull/133966
>From 4c71a107d174b24b4515554c4be57a2a318f5fb9 Mon Sep 17 00:00:00 2001
From: Andy Kaylor <akay...@nvidia.com>
Date: Tue, 1 Apr 2025 11:47:49 -0700
Subject: [PATCH] [CIR] Upstream support for logical not operations
When unary operation support was initially upstreamed, the cir.cast operation
hadn't been upstreamed yet, so logical not wasn't included. Since casts
have now been added, this change adds support for logical not.
---
.../CIR/Dialect/Builder/CIRBaseBuilder.h | 5 ++
clang/lib/CIR/CodeGen/CIRGenExprScalar.cpp | 31 ++++++-
.../CIR/Lowering/DirectToLLVM/LowerToLLVM.cpp | 9 +-
clang/test/CIR/CodeGen/unary.cpp | 90 +++++++++++++++++++
4 files changed, 132 insertions(+), 3 deletions(-)
diff --git a/clang/include/clang/CIR/Dialect/Builder/CIRBaseBuilder.h
b/clang/include/clang/CIR/Dialect/Builder/CIRBaseBuilder.h
index 51939e3af833d..c8fa2b4f2ea7f 100644
--- a/clang/include/clang/CIR/Dialect/Builder/CIRBaseBuilder.h
+++ b/clang/include/clang/CIR/Dialect/Builder/CIRBaseBuilder.h
@@ -121,6 +121,11 @@ class CIRBaseBuilderTy : public mlir::OpBuilder {
return cir::BoolAttr::get(getContext(), getBoolTy(), state);
}
+ mlir::Value createNot(mlir::Value value) {
+ return create<cir::UnaryOp>(value.getLoc(), value.getType(),
+ cir::UnaryOpKind::Not, value);
+ }
+
/// Create a do-while operation.
cir::DoWhileOp createDoWhile(
mlir::Location loc,
diff --git a/clang/lib/CIR/CodeGen/CIRGenExprScalar.cpp
b/clang/lib/CIR/CodeGen/CIRGenExprScalar.cpp
index 6289a8f1d2ed7..3863d21487531 100644
--- a/clang/lib/CIR/CodeGen/CIRGenExprScalar.cpp
+++ b/clang/lib/CIR/CodeGen/CIRGenExprScalar.cpp
@@ -488,6 +488,8 @@ class ScalarExprEmitter : public
StmtVisitor<ScalarExprEmitter, mlir::Value> {
return emitUnaryOp(e, cir::UnaryOpKind::Not, op);
}
+ mlir::Value VisitUnaryLNot(const UnaryOperator *e);
+
/// Emit a conversion from the specified type to the specified destination
/// type, both of which are CIR scalar types.
/// TODO: do we need ScalarConversionOpts here? Should be done in another
@@ -1315,7 +1317,7 @@ mlir::Value ScalarExprEmitter::VisitCastExpr(CastExpr
*ce) {
"fixed point casts");
return {};
}
- cgf.getCIRGenModule().errorNYI(subExpr->getSourceRange(), "fp options");
+ assert(!cir::MissingFeatures::cgFPOptionsRAII());
return emitScalarConversion(Visit(subExpr), subExpr->getType(), destTy,
ce->getExprLoc());
}
@@ -1353,6 +1355,33 @@ mlir::Value
CIRGenFunction::emitScalarConversion(mlir::Value src,
.emitScalarConversion(src, srcTy, dstTy, loc);
}
+mlir::Value ScalarExprEmitter::VisitUnaryLNot(const UnaryOperator *e) {
+ // Perform vector logical not on comparison with zero vector.
+ if (e->getType()->isVectorType() &&
+ e->getType()->castAs<VectorType>()->getVectorKind() ==
+ VectorKind::Generic) {
+ assert(!cir::MissingFeatures::vectorType());
+ cgf.cgm.errorNYI(e->getSourceRange(), "vector logical not");
+ return {};
+ }
+
+ // Compare operand to zero.
+ mlir::Value boolVal = cgf.evaluateExprAsBool(e->getSubExpr());
+
+ // Invert value.
+ boolVal = builder.createNot(boolVal);
+
+ // ZExt result to the expr type.
+ mlir::Type dstTy = cgf.convertType(e->getType());
+ if (mlir::isa<cir::IntType>(dstTy))
+ return builder.createBoolToInt(boolVal, dstTy);
+ if (mlir::isa<cir::BoolType>(dstTy))
+ return boolVal;
+
+ cgf.cgm.errorNYI("destination type for logical-not unary operator is NYI");
+ return {};
+}
+
/// Return the size or alignment of the type of argument of the sizeof
/// expression as an integer.
mlir::Value ScalarExprEmitter::VisitUnaryExprOrTypeTraitExpr(
diff --git a/clang/lib/CIR/Lowering/DirectToLLVM/LowerToLLVM.cpp
b/clang/lib/CIR/Lowering/DirectToLLVM/LowerToLLVM.cpp
index 1c455039269b9..81b80e2e4eafb 100644
--- a/clang/lib/CIR/Lowering/DirectToLLVM/LowerToLLVM.cpp
+++ b/clang/lib/CIR/Lowering/DirectToLLVM/LowerToLLVM.cpp
@@ -501,9 +501,14 @@ mlir::LogicalResult
CIRToLLVMCastOpLowering::matchAndRewrite(
assert(!MissingFeatures::cxxABI());
assert(!MissingFeatures::dataMemberType());
break;
- case cir::CastKind::ptr_to_bool:
- assert(!cir::MissingFeatures::opCmp());
+ case cir::CastKind::ptr_to_bool: {
+ mlir::Value llvmSrcVal = adaptor.getOperands().front();
+ mlir::Value zeroPtr = rewriter.create<mlir::LLVM::ZeroOp>(
+ castOp.getLoc(), llvmSrcVal.getType());
+ rewriter.replaceOpWithNewOp<mlir::LLVM::ICmpOp>(
+ castOp, mlir::LLVM::ICmpPredicate::ne, llvmSrcVal, zeroPtr);
break;
+ }
case cir::CastKind::address_space: {
mlir::Type dstTy = castOp.getType();
mlir::Value llvmSrcVal = adaptor.getOperands().front();
diff --git a/clang/test/CIR/CodeGen/unary.cpp b/clang/test/CIR/CodeGen/unary.cpp
index a6405e653a07c..b0a315a18f8fb 100644
--- a/clang/test/CIR/CodeGen/unary.cpp
+++ b/clang/test/CIR/CodeGen/unary.cpp
@@ -466,3 +466,93 @@ _Float16 fp16UMinus(_Float16 f) {
// OGCG: %[[PROMOTED:.*]] = fpext half %[[F_LOAD]] to float
// OGCG: %[[RESULT:.*]] = fneg float %[[PROMOTED]]
// OGCG: %[[UNPROMOTED:.*]] = fptrunc float %[[RESULT]] to half
+
+void test_logical_not() {
+ int a = 5;
+ a = !a;
+ bool b = false;
+ b = !b;
+ float c = 2.0f;
+ c = !c;
+ int *p = 0;
+ b = !p;
+ double d = 3.0;
+ b = !d;
+}
+
+// CHECK: cir.func @test_logical_not()
+// CHECK: %[[A:.*]] = cir.load %[[A_ADDR:.*]] : !cir.ptr<!s32i>, !s32i
+// CHECK: %[[A_BOOL:.*]] = cir.cast(int_to_bool, %[[A]] : !s32i), !cir.bool
+// CHECK: %[[A_NOT:.*]] = cir.unary(not, %[[A_BOOL]]) : !cir.bool, !cir.bool
+// CHECK: %[[A_CAST:.*]] = cir.cast(bool_to_int, %[[A_NOT]] : !cir.bool),
!s32i
+// CHECK: cir.store %[[A_CAST]], %[[A_ADDR]] : !s32i, !cir.ptr<!s32i>
+// CHECK: %[[B:.*]] = cir.load %[[B_ADDR:.*]] : !cir.ptr<!cir.bool>,
!cir.bool
+// CHECK: %[[B_NOT:.*]] = cir.unary(not, %[[B]]) : !cir.bool, !cir.bool
+// CHECK: cir.store %[[B_NOT]], %[[B_ADDR]] : !cir.bool, !cir.ptr<!cir.bool>
+// CHECK: %[[C:.*]] = cir.load %[[C_ADDR:.*]] : !cir.ptr<!cir.float>,
!cir.float
+// CHECK: %[[C_BOOL:.*]] = cir.cast(float_to_bool, %[[C]] : !cir.float),
!cir.bool
+// CHECK: %[[C_NOT:.*]] = cir.unary(not, %[[C_BOOL]]) : !cir.bool, !cir.bool
+// CHECK: %[[C_CAST:.*]] = cir.cast(bool_to_float, %[[C_NOT]] : !cir.bool),
!cir.float
+// CHECK: cir.store %[[C_CAST]], %[[C_ADDR]] : !cir.float,
!cir.ptr<!cir.float>
+// CHECK: %[[P:.*]] = cir.load %[[P_ADDR:.*]] : !cir.ptr<!cir.ptr<!s32i>>,
!cir.ptr<!s32i>
+// CHECK: %[[P_BOOL:.*]] = cir.cast(ptr_to_bool, %[[P]] : !cir.ptr<!s32i>),
!cir.bool
+// CHECK: %[[P_NOT:.*]] = cir.unary(not, %[[P_BOOL]]) : !cir.bool, !cir.bool
+// CHECK: cir.store %[[P_NOT]], %[[B_ADDR]] : !cir.bool, !cir.ptr<!cir.bool>
+// CHECK: %[[D:.*]] = cir.load %[[D_ADDR:.*]] : !cir.ptr<!cir.double>,
!cir.double
+// CHECK: %[[D_BOOL:.*]] = cir.cast(float_to_bool, %[[D]] : !cir.double),
!cir.bool
+// CHECK: %[[D_NOT:.*]] = cir.unary(not, %[[D_BOOL]]) : !cir.bool, !cir.bool
+// CHECK: cir.store %[[D_NOT]], %[[B_ADDR]] : !cir.bool, !cir.ptr<!cir.bool>
+
+// LLVM: define void @test_logical_not()
+// LLVM: %[[A:.*]] = load i32, ptr %[[A_ADDR:.*]], align 4
+// LLVM: %[[A_BOOL:.*]] = icmp ne i32 %[[A]], 0
+// LLVM: %[[A_NOT:.*]] = xor i1 %[[A_BOOL]], true
+// LLVM: %[[A_CAST:.*]] = zext i1 %[[A_NOT]] to i32
+// LLVM: store i32 %[[A_CAST]], ptr %[[A_ADDR]], align 4
+// LLVM: %[[B:.*]] = load i8, ptr %[[B_ADDR:.*]], align 1
+// LLVM: %[[B_BOOL:.*]] = trunc i8 %[[B]] to i1
+// LLVM: %[[B_NOT:.*]] = xor i1 %[[B_BOOL]], true
+// LLVM: %[[B_CAST:.*]] = zext i1 %[[B_NOT]] to i8
+// LLVM: store i8 %[[B_CAST]], ptr %[[B_ADDR]], align 1
+// LLVM: %[[C:.*]] = load float, ptr %[[C_ADDR:.*]], align 4
+// LLVM: %[[C_BOOL:.*]] = fcmp une float %[[C]], 0.000000e+00
+// LLVM: %[[C_NOT:.*]] = xor i1 %[[C_BOOL]], true
+// LLVM: %[[C_CAST:.*]] = uitofp i1 %[[C_NOT]] to float
+// LLVM: store float %[[C_CAST]], ptr %[[C_ADDR]], align 4
+// LLVM: %[[P:.*]] = load ptr, ptr %[[P_ADDR:.*]], align 8
+// LLVM: %[[P_BOOL:.*]] = icmp ne ptr %[[P]], null
+// LLVM: %[[P_NOT:.*]] = xor i1 %[[P_BOOL]], true
+// LLVM: %[[P_CAST:.*]] = zext i1 %[[P_NOT]] to i8
+// LLVM: store i8 %[[P_CAST]], ptr %[[B_ADDR]], align 1
+// LLVM: %[[D:.*]] = load double, ptr %[[D_ADDR:.*]], align 8
+// LLVM: %[[D_BOOL:.*]] = fcmp une double %[[D]], 0.000000e+00
+// LLVM: %[[D_NOT:.*]] = xor i1 %[[D_BOOL]], true
+// LLVM: %[[D_CAST:.*]] = zext i1 %[[D_NOT]] to i8
+// LLVM: store i8 %[[D_CAST]], ptr %[[B_ADDR]], align 1
+
+// OGCG: define{{.*}} void @_Z16test_logical_notv()
+// OGCG: %[[A:.*]] = load i32, ptr %[[A_ADDR:.*]], align 4
+// OGCG: %[[A_BOOL:.*]] = icmp ne i32 %[[A]], 0
+// OGCG: %[[A_NOT:.*]] = xor i1 %[[A_BOOL]], true
+// OGCG: %[[A_CAST:.*]] = zext i1 %[[A_NOT]] to i32
+// OGCG: store i32 %[[A_CAST]], ptr %[[A_ADDR]], align 4
+// OGCG: %[[B:.*]] = load i8, ptr %[[B_ADDR:.*]], align 1
+// OGCG: %[[B_BOOL:.*]] = trunc i8 %[[B]] to i1
+// OGCG: %[[B_NOT:.*]] = xor i1 %[[B_BOOL]], true
+// OGCG: %[[B_CAST:.*]] = zext i1 %[[B_NOT]] to i8
+// OGCG: store i8 %[[B_CAST]], ptr %[[B_ADDR]], align 1
+// OGCG: %[[C:.*]] = load float, ptr %[[C_ADDR:.*]], align 4
+// OGCG: %[[C_BOOL:.*]] = fcmp une float %[[C]], 0.000000e+00
+// OGCG: %[[C_NOT:.*]] = xor i1 %[[C_BOOL]], true
+// OGCG: %[[C_CAST:.*]] = uitofp i1 %[[C_NOT]] to float
+// OGCG: store float %[[C_CAST]], ptr %[[C_ADDR]], align 4
+// OGCG: %[[P:.*]] = load ptr, ptr %[[P_ADDR:.*]], align 8
+// OGCG: %[[P_BOOL:.*]] = icmp ne ptr %[[P]], null
+// OGCG: %[[P_NOT:.*]] = xor i1 %[[P_BOOL]], true
+// OGCG: %[[P_CAST:.*]] = zext i1 %[[P_NOT]] to i8
+// OGCG: store i8 %[[P_CAST]], ptr %[[B_ADDR]], align 1
+// OGCG: %[[D:.*]] = load double, ptr %[[D_ADDR:.*]], align 8
+// OGCG: %[[D_BOOL:.*]] = fcmp une double %[[D]], 0.000000e+00
+// OGCG: %[[D_NOT:.*]] = xor i1 %[[D_BOOL]], true
+// OGCG: %[[D_CAST:.*]] = zext i1 %[[D_NOT]] to i8
+// OGCG: store i8 %[[D_CAST]], ptr %[[B_ADDR]], align 1
_______________________________________________
cfe-commits mailing list
cfe-commits@lists.llvm.org
https://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits
