diff options
Diffstat (limited to 'mlir/lib/Dialect/GPU/Transforms/KernelOutlining.cpp')
| -rw-r--r-- | mlir/lib/Dialect/GPU/Transforms/KernelOutlining.cpp | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/mlir/lib/Dialect/GPU/Transforms/KernelOutlining.cpp b/mlir/lib/Dialect/GPU/Transforms/KernelOutlining.cpp index 6a7cd290dd2..2d00ac03d33 100644 --- a/mlir/lib/Dialect/GPU/Transforms/KernelOutlining.cpp +++ b/mlir/lib/Dialect/GPU/Transforms/KernelOutlining.cpp @@ -22,10 +22,10 @@ using namespace mlir; template <typename OpTy> static void createForAllDimensions(OpBuilder &builder, Location loc, - SmallVectorImpl<ValuePtr> &values) { + SmallVectorImpl<Value> &values) { for (StringRef dim : {"x", "y", "z"}) { - ValuePtr v = builder.create<OpTy>(loc, builder.getIndexType(), - builder.getStringAttr(dim)); + Value v = builder.create<OpTy>(loc, builder.getIndexType(), + builder.getStringAttr(dim)); values.push_back(v); } } @@ -37,7 +37,7 @@ static void injectGpuIndexOperations(Location loc, Region &body) { OpBuilder builder(loc->getContext()); Block &firstBlock = body.front(); builder.setInsertionPointToStart(&firstBlock); - SmallVector<ValuePtr, 12> indexOps; + SmallVector<Value, 12> indexOps; createForAllDimensions<gpu::BlockIdOp>(builder, loc, indexOps); createForAllDimensions<gpu::ThreadIdOp>(builder, loc, indexOps); createForAllDimensions<gpu::GridDimOp>(builder, loc, indexOps); @@ -60,7 +60,7 @@ static gpu::LaunchFuncOp inlineBeneficiaryOps(gpu::GPUFuncOp kernelFunc, gpu::LaunchFuncOp launch) { OpBuilder kernelBuilder(kernelFunc.getBody()); auto &firstBlock = kernelFunc.getBody().front(); - SmallVector<ValuePtr, 8> newLaunchArgs; + SmallVector<Value, 8> newLaunchArgs; BlockAndValueMapping map; for (int i = 0, e = launch.getNumKernelOperands(); i < e; ++i) { map.map(launch.getKernelOperand(i), kernelFunc.getArgument(i)); @@ -73,7 +73,7 @@ static gpu::LaunchFuncOp inlineBeneficiaryOps(gpu::GPUFuncOp kernelFunc, } // Only inline operations that do not create new arguments. if (!llvm::all_of(operandOp->getOperands(), - [map](ValuePtr value) { return map.contains(value); })) { + [map](Value value) { return map.contains(value); })) { continue; } auto clone = kernelBuilder.clone(*operandOp, map); |
