diff options
Diffstat (limited to 'mlir/lib/Conversion/GPUToLLVMSPV/GPUToLLVMSPV.cpp')
-rw-r--r-- | mlir/lib/Conversion/GPUToLLVMSPV/GPUToLLVMSPV.cpp | 28 |
1 files changed, 14 insertions, 14 deletions
diff --git a/mlir/lib/Conversion/GPUToLLVMSPV/GPUToLLVMSPV.cpp b/mlir/lib/Conversion/GPUToLLVMSPV/GPUToLLVMSPV.cpp index 8b6b553..c2363a1 100644 --- a/mlir/lib/Conversion/GPUToLLVMSPV/GPUToLLVMSPV.cpp +++ b/mlir/lib/Conversion/GPUToLLVMSPV/GPUToLLVMSPV.cpp @@ -54,8 +54,8 @@ static LLVM::LLVMFuncOp lookupOrCreateSPIRVFn(Operation *symbolTable, SymbolTable::lookupSymbolIn(symbolTable, name)); if (!func) { OpBuilder b(symbolTable->getRegion(0)); - func = b.create<LLVM::LLVMFuncOp>( - symbolTable->getLoc(), name, + func = LLVM::LLVMFuncOp::create( + b, symbolTable->getLoc(), name, LLVM::LLVMFunctionType::get(resultType, paramTypes)); func.setCConv(LLVM::cconv::CConv::SPIR_FUNC); func.setNoUnwind(true); @@ -79,7 +79,7 @@ static LLVM::CallOp createSPIRVBuiltinCall(Location loc, ConversionPatternRewriter &rewriter, LLVM::LLVMFuncOp func, ValueRange args) { - auto call = rewriter.create<LLVM::CallOp>(loc, func, args); + auto call = LLVM::CallOp::create(rewriter, loc, func, args); call.setCConv(func.getCConv()); call.setConvergentAttr(func.getConvergentAttr()); call.setNoUnwindAttr(func.getNoUnwindAttr()); @@ -121,7 +121,7 @@ struct GPUBarrierConversion final : ConvertOpToLLVMPattern<gpu::BarrierOp> { constexpr int64_t localMemFenceFlag = 1; Location loc = op->getLoc(); Value flag = - rewriter.create<LLVM::ConstantOp>(loc, flagTy, localMemFenceFlag); + LLVM::ConstantOp::create(rewriter, loc, flagTy, localMemFenceFlag); rewriter.replaceOp(op, createSPIRVBuiltinCall(loc, rewriter, func, flag)); return success(); } @@ -162,8 +162,8 @@ struct LaunchConfigConversion : ConvertToLLVMPattern { Location loc = op->getLoc(); gpu::Dimension dim = getDimension(op); - Value dimVal = rewriter.create<LLVM::ConstantOp>(loc, dimTy, - static_cast<int64_t>(dim)); + Value dimVal = LLVM::ConstantOp::create(rewriter, loc, dimTy, + static_cast<int64_t>(dim)); rewriter.replaceOp(op, createSPIRVBuiltinCall(loc, rewriter, func, dimVal)); return success(); } @@ -291,13 +291,13 @@ struct GPUShuffleConversion final : ConvertOpToLLVMPattern<gpu::ShuffleOp> { ConversionPatternRewriter &rewriter) { return TypeSwitch<Type, Value>(oldVal.getType()) .Case([&](BFloat16Type) { - return rewriter.create<LLVM::BitcastOp>(loc, rewriter.getI16Type(), - oldVal); + return LLVM::BitcastOp::create(rewriter, loc, rewriter.getI16Type(), + oldVal); }) .Case([&](IntegerType intTy) -> Value { if (intTy.getWidth() == 1) - return rewriter.create<LLVM::ZExtOp>(loc, rewriter.getI8Type(), - oldVal); + return LLVM::ZExtOp::create(rewriter, loc, rewriter.getI8Type(), + oldVal); return oldVal; }) .Default(oldVal); @@ -308,11 +308,11 @@ struct GPUShuffleConversion final : ConvertOpToLLVMPattern<gpu::ShuffleOp> { ConversionPatternRewriter &rewriter) { return TypeSwitch<Type, Value>(newTy) .Case([&](BFloat16Type) { - return rewriter.create<LLVM::BitcastOp>(loc, newTy, oldVal); + return LLVM::BitcastOp::create(rewriter, loc, newTy, oldVal); }) .Case([&](IntegerType intTy) -> Value { if (intTy.getWidth() == 1) - return rewriter.create<LLVM::TruncOp>(loc, newTy, oldVal); + return LLVM::TruncOp::create(rewriter, loc, newTy, oldVal); return oldVal; }) .Default(oldVal); @@ -349,7 +349,7 @@ struct GPUShuffleConversion final : ConvertOpToLLVMPattern<gpu::ShuffleOp> { bitcastOrTruncAfterShuffle(result, op.getType(0), loc, rewriter); Value trueVal = - rewriter.create<LLVM::ConstantOp>(loc, rewriter.getI1Type(), true); + LLVM::ConstantOp::create(rewriter, loc, rewriter.getI1Type(), true); rewriter.replaceOp(op, {resultOrConversion, trueVal}); return success(); } @@ -426,7 +426,7 @@ struct GPUSubgroupOpConversion final : ConvertOpToLLVMPattern<SubgroupOp> { if (indexTy.getIntOrFloatBitWidth() < resultTy.getIntOrFloatBitWidth()) { return failure(); } - result = rewriter.create<LLVM::ZExtOp>(loc, indexTy, result); + result = LLVM::ZExtOp::create(rewriter, loc, indexTy, result); } rewriter.replaceOp(op, result); |