Skip to content

Commit 5ed5160

Browse files
cferry-AMDcferry-AMD
committed
Use new types in EmitC, lower index_cast
1 parent 519175c commit 5ed5160

File tree

3 files changed

+120
-20
lines changed

3 files changed

+120
-20
lines changed

‎mlir/lib/Conversion/ArithToEmitC/ArithToEmitC.cpp

Lines changed: 40 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515

1616
#include "mlir/Dialect/Arith/IR/Arith.h"
1717
#include "mlir/Dialect/EmitC/IR/EmitC.h"
18+
#include "mlir/Dialect/EmitC/Transforms/TypeConversions.h"
1819
#include "mlir/IR/BuiltinAttributes.h"
1920
#include "mlir/IR/BuiltinTypes.h"
2021
#include "mlir/Support/LogicalResult.h"
@@ -36,8 +37,11 @@ class ArithConstantOpConversionPattern
3637
matchAndRewrite(arith::ConstantOp arithConst,
3738
arith::ConstantOp::Adaptor adaptor,
3839
ConversionPatternRewriter &rewriter) const override {
39-
rewriter.replaceOpWithNewOp<emitc::ConstantOp>(
40-
arithConst, arithConst.getType(), adaptor.getValue());
40+
Type newTy = this->getTypeConverter()->convertType(arithConst.getType());
41+
if (!newTy)
42+
return rewriter.notifyMatchFailure(arithConst, "type conversion failed");
43+
rewriter.replaceOpWithNewOp<emitc::ConstantOp>(arithConst, newTy,
44+
adaptor.getValue());
4145
return success();
4246
}
4347
};
@@ -52,6 +56,12 @@ Type adaptIntegralTypeSignedness(Type ty, bool needsUnsigned) {
5256
return IntegerType::get(ty.getContext(), ty.getIntOrFloatBitWidth(),
5357
signedness);
5458
}
59+
} else if (emitc::isPointerWideType(ty)) {
60+
if (isa<emitc::SizeTType>(ty) != needsUnsigned) {
61+
if (needsUnsigned)
62+
return emitc::SizeTType::get(ty.getContext());
63+
return emitc::PtrDiffTType::get(ty.getContext());
64+
}
5565
}
5666
return ty;
5767
}
@@ -264,8 +274,9 @@ class CmpIOpConversion : public OpConversionPattern<arith::CmpIOp> {
264274
ConversionPatternRewriter &rewriter) const override {
265275

266276
Type type = adaptor.getLhs().getType();
267-
if (!isa_and_nonnull<IntegerType, IndexType>(type)) {
268-
return rewriter.notifyMatchFailure(op, "expected integer or index type");
277+
if (type && !(isa<IntegerType>(type) || emitc::isPointerWideType(type))) {
278+
return rewriter.notifyMatchFailure(
279+
op, "expected integer or size_t/ssize_t type");
269280
}
270281

271282
bool needsUnsigned = needsUnsignedCmp(op.getPredicate());
@@ -318,17 +329,21 @@ class CastConversion : public OpConversionPattern<ArithOp> {
318329
ConversionPatternRewriter &rewriter) const override {
319330

320331
Type opReturnType = this->getTypeConverter()->convertType(op.getType());
321-
if (!isa_and_nonnull<IntegerType>(opReturnType))
322-
return rewriter.notifyMatchFailure(op, "expected integer result type");
332+
if (opReturnType && !(isa_and_nonnull<IntegerType>(opReturnType) ||
333+
emitc::isPointerWideType(opReturnType)))
334+
return rewriter.notifyMatchFailure(
335+
op, "expected integer or size_t/ssize_t result type");
323336

324337
if (adaptor.getOperands().size() != 1) {
325338
return rewriter.notifyMatchFailure(
326339
op, "CastConversion only supports unary ops");
327340
}
328341

329342
Type operandType = adaptor.getIn().getType();
330-
if (!isa_and_nonnull<IntegerType>(operandType))
331-
return rewriter.notifyMatchFailure(op, "expected integer operand type");
343+
if (operandType && !(isa_and_nonnull<IntegerType>(operandType) ||
344+
emitc::isPointerWideType(operandType)))
345+
return rewriter.notifyMatchFailure(
346+
op, "expected integer or size_t/ssize_t operand type");
332347

333348
// Signed (sign-extending) casts from i1 are not supported.
334349
if (operandType.isInteger(1) && !castToUnsigned)
@@ -339,8 +354,11 @@ class CastConversion : public OpConversionPattern<ArithOp> {
339354
// equivalent to (v != 0). Implementing as (bool)(v & 0x01) gives
340355
// truncation.
341356
if (opReturnType.isInteger(1)) {
357+
Type attrType = (emitc::isPointerWideType(operandType))
358+
? rewriter.getIndexType()
359+
: operandType;
342360
auto constOne = rewriter.create<emitc::ConstantOp>(
343-
op.getLoc(), operandType, rewriter.getIntegerAttr(operandType, 1));
361+
op.getLoc(), operandType, rewriter.getIntegerAttr(attrType, 1));
344362
auto oneAndOperand = rewriter.create<emitc::BitwiseAndOp>(
345363
op.getLoc(), operandType, adaptor.getIn(), constOne);
346364
rewriter.replaceOpWithNewOp<emitc::CastOp>(op, opReturnType,
@@ -393,7 +411,11 @@ class ArithOpConversion final : public OpConversionPattern<ArithOp> {
393411
matchAndRewrite(ArithOp arithOp, typename ArithOp::Adaptor adaptor,
394412
ConversionPatternRewriter &rewriter) const override {
395413

396-
rewriter.template replaceOpWithNewOp<EmitCOp>(arithOp, arithOp.getType(),
414+
Type newTy = this->getTypeConverter()->convertType(arithOp.getType());
415+
if (!newTy)
416+
return rewriter.notifyMatchFailure(arithOp,
417+
"converting result type failed");
418+
rewriter.template replaceOpWithNewOp<EmitCOp>(arithOp, newTy,
397419
adaptor.getOperands());
398420

399421
return success();
@@ -410,8 +432,10 @@ class IntegerOpConversion final : public OpConversionPattern<ArithOp> {
410432
ConversionPatternRewriter &rewriter) const override {
411433

412434
Type type = this->getTypeConverter()->convertType(op.getType());
413-
if (!isa_and_nonnull<IntegerType, IndexType>(type)) {
414-
return rewriter.notifyMatchFailure(op, "expected integer type");
435+
if (type && !(isa_and_nonnull<IntegerType>(type) ||
436+
emitc::isPointerWideType(type))) {
437+
return rewriter.notifyMatchFailure(
438+
op, "expected integer or size_t/ssize_t/ptrdiff_t type");
415439
}
416440

417441
if (type.isInteger(1)) {
@@ -606,6 +630,8 @@ void mlir::populateArithToEmitCPatterns(TypeConverter &typeConverter,
606630
RewritePatternSet &patterns) {
607631
MLIRContext *ctx = patterns.getContext();
608632

633+
mlir::populateEmitCSizeTTypeConversions(typeConverter);
634+
609635
// clang-format off
610636
patterns.add<
611637
ArithConstantOpConversionPattern,
@@ -629,6 +655,8 @@ void mlir::populateArithToEmitCPatterns(TypeConverter &typeConverter,
629655
UnsignedCastConversion<arith::TruncIOp>,
630656
SignedCastConversion<arith::ExtSIOp>,
631657
UnsignedCastConversion<arith::ExtUIOp>,
658+
SignedCastConversion<arith::IndexCastOp>,
659+
UnsignedCastConversion<arith::IndexCastUIOp>,
632660
ItoFCastOpConversion<arith::SIToFPOp>,
633661
ItoFCastOpConversion<arith::UIToFPOp>,
634662
FtoICastOpConversion<arith::FPToSIOp>,

‎mlir/lib/Conversion/ArithToEmitC/CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@ add_mlir_conversion_library(MLIRArithToEmitC
1111
LINK_LIBS PUBLIC
1212
MLIRArithDialect
1313
MLIREmitCDialect
14+
MLIREmitCTransforms
1415
MLIRPass
1516
MLIRTransformUtils
1617
)

‎mlir/test/Conversion/ArithToEmitC/arith-to-emitc.mlir

Lines changed: 79 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,8 @@
33
// CHECK-LABEL: arith_constants
44
func.func @arith_constants() {
55
// CHECK: emitc.constant
6-
// CHECK-SAME: value = 0 : index
6+
// CHECK-SAME: value = 0
7+
// CHECK-SAME: () -> !emitc.size_t
78
%c_index = arith.constant 0 : index
89
// CHECK: emitc.constant
910
// CHECK-SAME: value = 0 : i32
@@ -75,13 +76,18 @@ func.func @arith_integer_ops_signed_nsw(%arg0: i32, %arg1: i32) {
7576
// -----
7677

7778
// CHECK-LABEL: arith_index
78-
func.func @arith_index(%arg0: index, %arg1: index) {
79-
// CHECK: emitc.add %arg0, %arg1 : (index, index) -> index
80-
%0 = arith.addi %arg0, %arg1 : index
81-
// CHECK: emitc.sub %arg0, %arg1 : (index, index) -> index
82-
%1 = arith.subi %arg0, %arg1 : index
83-
// CHECK: emitc.mul %arg0, %arg1 : (index, index) -> index
84-
%2 = arith.muli %arg0, %arg1 : index
79+
func.func @arith_index(%arg0: i32, %arg1: i32) {
80+
// CHECK: %[[CST0:.*]] = emitc.cast %{{.*}} : {{.*}} to !emitc.size_t
81+
%cst0 = arith.index_cast %arg0 : i32 to index
82+
// CHECK: %[[CST1:.*]] = emitc.cast %{{.*}} : {{.*}} to !emitc.size_t
83+
%cst1 = arith.index_cast %arg1 : i32 to index
84+
85+
// CHECK: emitc.add %[[CST0]], %[[CST1]] : (!emitc.size_t, !emitc.size_t) -> !emitc.size_t
86+
%0 = arith.addi %cst0, %cst1 : index
87+
// CHECK: emitc.sub %[[CST0]], %[[CST1]] : (!emitc.size_t, !emitc.size_t) -> !emitc.size_t
88+
%1 = arith.subi %cst0, %cst1 : index
89+
// CHECK: emitc.mul %[[CST0]], %[[CST1]] : (!emitc.size_t, !emitc.size_t) -> !emitc.size_t
90+
%2 = arith.muli %cst0, %cst1 : index
8591

8692
return
8793
}
@@ -420,6 +426,27 @@ func.func @arith_cmpi_predicates(%arg0: i32, %arg1: i32) {
420426
return
421427
}
422428

429+
func.func @arith_cmpi_index(%arg0: i32, %arg1: i32) -> i1 {
430+
// CHECK-LABEL: arith_cmpi_index
431+
432+
// CHECK: %[[Cst0:.*]] = emitc.cast %{{.*}} : {{.*}} to !emitc.size_t
433+
%idx0 = arith.index_cast %arg0 : i32 to index
434+
// CHECK: %[[Cst1:.*]] = emitc.cast %{{.*}} : {{.*}} to !emitc.size_t
435+
%idx1 = arith.index_cast %arg0 : i32 to index
436+
437+
// CHECK-DAG: [[ULT:[^ ]*]] = emitc.cmp lt, %[[Cst0]], %[[Cst1]] : (!emitc.size_t, !emitc.size_t) -> i1
438+
%ult = arith.cmpi ult, %idx0, %idx1 : index
439+
440+
// CHECK-DAG: %[[CastArg0:[^ ]*]] = emitc.cast %[[Cst0]] : !emitc.size_t to !emitc.ptrdiff_t
441+
// CHECK-DAG: %[[CastArg1:[^ ]*]] = emitc.cast %[[Cst1]] : !emitc.size_t to !emitc.ptrdiff_t
442+
// CHECK-DAG: %[[SLT:[^ ]*]] = emitc.cmp lt, %[[CastArg0]], %[[CastArg1]] : (!emitc.ptrdiff_t, !emitc.ptrdiff_t) -> i1
443+
%slt = arith.cmpi slt, %idx0, %idx1 : index
444+
445+
// CHECK: return %[[SLT]]
446+
return %slt: i1
447+
}
448+
449+
423450
// -----
424451

425452
func.func @arith_negf(%arg0: f32) -> f32 {
@@ -536,3 +563,47 @@ func.func @arith_extui_i1_to_i32(%arg0: i1) {
536563
%idx = arith.extui %arg0 : i1 to i32
537564
return
538565
}
566+
567+
// -----
568+
569+
func.func @arith_index_cast(%arg0: i32) -> i32 {
570+
// CHECK-LABEL: arith_index_cast
571+
// CHECK-SAME: (%[[Arg0:[^ ]*]]: i32)
572+
// CHECK: %[[Conv0:.*]] = emitc.cast %[[Arg0]] : i32 to !emitc.ptrdiff_t
573+
// CHECK: %[[Conv1:.*]] = emitc.cast %[[Conv0]] : !emitc.ptrdiff_t to !emitc.size_t
574+
%idx = arith.index_cast %arg0 : i32 to index
575+
// CHECK: %[[Conv2:.*]] = emitc.cast %[[Conv1]] : !emitc.size_t to !emitc.ptrdiff_t
576+
// CHECK: %[[Conv3:.*]] = emitc.cast %[[Conv2]] : !emitc.ptrdiff_t to i32
577+
%int = arith.index_cast %idx : index to i32
578+
579+
// CHECK: %[[Const:.*]] = "emitc.constant"
580+
// CHECK-SAME: value = 1
581+
// CHECK-SAME: () -> !emitc.size_t
582+
// CHECK: %[[AndOne:.*]] = emitc.bitwise_and %[[Conv1]], %[[Const]] : (!emitc.size_t, !emitc.size_t) -> !emitc.size_t
583+
// CHECK: %[[Conv4:.*]] = emitc.cast %[[AndOne]] : !emitc.size_t to i1
584+
%bool = arith.index_cast %idx : index to i1
585+
586+
return %int : i32
587+
}
588+
589+
// -----
590+
591+
func.func @arith_index_castui(%arg0: i32) -> i32 {
592+
// CHECK-LABEL: arith_index_castui
593+
// CHECK-SAME: (%[[Arg0:[^ ]*]]: i32)
594+
// CHECK: %[[Conv0:.*]] = emitc.cast %[[Arg0]] : i32 to ui32
595+
// CHECK: %[[Conv1:.*]] = emitc.cast %[[Conv0]] : ui32 to !emitc.size_t
596+
%idx = arith.index_castui %arg0 : i32 to index
597+
// CHECK: %[[Conv2:.*]] = emitc.cast %[[Conv1]] : !emitc.size_t to ui32
598+
// CHECK: %[[Conv3:.*]] = emitc.cast %[[Conv2]] : ui32 to i32
599+
%int = arith.index_castui %idx : index to i32
600+
601+
// CHECK: %[[Const:.*]] = "emitc.constant"
602+
// CHECK-SAME: value = 1
603+
// CHECK-SAME: () -> !emitc.size_t
604+
// CHECK: %[[AndOne:.*]] = emitc.bitwise_and %[[Conv1]], %[[Const]] : (!emitc.size_t, !emitc.size_t) -> !emitc.size_t
605+
// CHECK: %[[Conv4:.*]] = emitc.cast %[[AndOne]] : !emitc.size_t to i1
606+
%bool = arith.index_castui %idx : index to i1
607+
608+
return %int : i32
609+
}

0 commit comments

Comments
 (0)