diff options
| author | River Riddle <riverriddle@google.com> | 2019-12-07 10:35:01 -0800 |
|---|---|---|
| committer | A. Unique TensorFlower <gardener@tensorflow.org> | 2019-12-07 10:35:41 -0800 |
| commit | d6ee6a031063cb99ca9828f1698a60536ffbb38a (patch) | |
| tree | b8b1b6ecf500d93818366914f56dc818054339f3 /mlir/lib/Dialect/GPU/Transforms | |
| parent | 9d1a0c72b4ae54b97809966257bd1b9cb3140dfe (diff) | |
| download | bcm5719-llvm-d6ee6a031063cb99ca9828f1698a60536ffbb38a.tar.gz bcm5719-llvm-d6ee6a031063cb99ca9828f1698a60536ffbb38a.zip | |
Update the builder API to take ValueRange instead of ArrayRef<Value *>
This allows for users to provide operand_range and result_range in builder.create<> calls, instead of requiring an explicit copy into a separate data structure like SmallVector/std::vector.
PiperOrigin-RevId: 284360710
Diffstat (limited to 'mlir/lib/Dialect/GPU/Transforms')
| -rw-r--r-- | mlir/lib/Dialect/GPU/Transforms/KernelOutlining.cpp | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/mlir/lib/Dialect/GPU/Transforms/KernelOutlining.cpp b/mlir/lib/Dialect/GPU/Transforms/KernelOutlining.cpp index fbcd18b9535..b466cc280df 100644 --- a/mlir/lib/Dialect/GPU/Transforms/KernelOutlining.cpp +++ b/mlir/lib/Dialect/GPU/Transforms/KernelOutlining.cpp @@ -133,11 +133,9 @@ static FuncOp outlineKernelFunc(gpu::LaunchOp launchOp) { // constant region arguments inlined. static void convertToLaunchFuncOp(gpu::LaunchOp &launchOp, FuncOp kernelFunc) { OpBuilder builder(launchOp); - SmallVector<Value *, 4> kernelOperandValues( - launchOp.getKernelOperandValues()); auto launchFuncOp = builder.create<gpu::LaunchFuncOp>( launchOp.getLoc(), kernelFunc, launchOp.getGridSizeOperandValues(), - launchOp.getBlockSizeOperandValues(), kernelOperandValues); + launchOp.getBlockSizeOperandValues(), launchOp.getKernelOperandValues()); inlineBeneficiaryOps(kernelFunc, launchFuncOp); launchOp.erase(); } |
