Magic-Mirror/llvm-capstone
1
0
Fork 0
mirror of https://github.com/capstone-engine/llvm-capstone.git synced 2025-01-14 12:12:07 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

[mlir][StandardToSPIRV] Handle i1 case for lowering std.zexti to SPIR-V.

Browse Source 
Differential Revision: https://reviews.llvm.org/D80965
This commit is contained in:
Hanhan Wang 2020-06-03 15:00:33 -07:00
parent 0e7c77053f
commit 0b025d2733
2 changed files with 36 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
mlir/lib/Conversion/StandardToSPIRV/ConvertStandardToSPIRV.cpp
View File

@ -445,6 +445,28 @@ public:
ConversionPatternRewriter &rewriter) const override;
};
/// Converts std.zexti to spv.Select if the type of source is i1.
class ZeroExtendI1Pattern final : public SPIRVOpLowering<ZeroExtendIOp> {
public:
using SPIRVOpLowering<ZeroExtendIOp>::SPIRVOpLowering;
LogicalResult
matchAndRewrite(ZeroExtendIOp op, ArrayRef<Value> operands,
ConversionPatternRewriter &rewriter) const override {
auto srcType = operands.front().getType();
if (!srcType.isSignlessInteger() || srcType.getIntOrFloatBitWidth() != 1)
return failure();
auto dstType = this->typeConverter.convertType(op.getResult().getType());
Location loc = op.getLoc();
Value zero = rewriter.create<ConstantIntOp>(loc, 0, dstType);
Value one = rewriter.create<ConstantIntOp>(loc, 1, dstType);
rewriter.template replaceOpWithNewOp<spirv::SelectOp>(
op, dstType, operands.front(), one, zero);
return success();
}
};
/// Converts type-casting standard operations to SPIR-V operations.
template <typename StdOp, typename SPIRVOp>
class TypeCastingOpPattern final : public SPIRVOpLowering<StdOp> {
@ -455,9 +477,12 @@ public:
matchAndRewrite(StdOp operation, ArrayRef<Value> operands,
ConversionPatternRewriter &rewriter) const override {
assert(operands.size() == 1);
auto srcType = operands.front().getType();
if (srcType.isSignlessInteger() && srcType.getIntOrFloatBitWidth() == 1)
return failure();
auto dstType =
this->typeConverter.convertType(operation.getResult().getType());
if (dstType == operands.front().getType()) {
if (dstType == srcType) {
// Due to type conversion, we are seeing the same source and target type.
// Then we can just erase this operation by forwarding its operand.
rewriter.replaceOp(operation, operands.front());
@ -1012,7 +1037,7 @@ void populateStandardToSPIRVPatterns(MLIRContext *context,
BoolCmpIOpPattern, ConstantCompositeOpPattern, ConstantScalarOpPattern,
CmpFOpPattern, CmpIOpPattern, IntLoadOpPattern, LoadOpPattern,
ReturnOpPattern, SelectOpPattern, IntStoreOpPattern, StoreOpPattern,
TypeCastingOpPattern<IndexCastOp, spirv::SConvertOp>,
ZeroExtendI1Pattern, TypeCastingOpPattern<IndexCastOp, spirv::SConvertOp>,
TypeCastingOpPattern<SIToFPOp, spirv::ConvertSToFOp>,
TypeCastingOpPattern<ZeroExtendIOp, spirv::UConvertOp>,
TypeCastingOpPattern<TruncateIOp, spirv::SConvertOp>,

9
mlir/test/Conversion/StandardToSPIRV/std-ops-to-spirv.mlir
View File

@ -564,6 +564,15 @@ func @zexti2(%arg0 : i32) -> i64 {
return %0 : i64
}
// CHECK-LABEL: @zexti3
func @zexti3(%arg0 : i1) -> i32 {
// CHECK: %[[ZERO:.+]] = spv.constant 0 : i32
// CHECK: %[[ONE:.+]] = spv.constant 1 : i32
// CHECK: spv.Select %{{.*}}, %[[ONE]], %[[ZERO]] : i1, i32
%0 = std.zexti %arg0 : i1 to i32
return %0 : i32
}
// CHECK-LABEL: @trunci1
func @trunci1(%arg0 : i64) -> i16 {
// CHECK: spv.SConvert %{{.*}} : i64 to i16