diff options
| author | River Riddle <riverriddle@google.com> | 2019-10-30 11:13:52 -0700 |
|---|---|---|
| committer | A. Unique TensorFlower <gardener@tensorflow.org> | 2019-10-30 11:14:30 -0700 |
| commit | 0568e952b6d1dc53f44c8eb5af167fc9d2e0bb34 (patch) | |
| tree | 8b4837d53c6d06c22f9bfbe54252568a61ef578f /mlir/lib/Dialect/SPIRV/SPIRVOps.cpp | |
| parent | cb40e36d3bb93e876ec6d05f128a23f26b952604 (diff) | |
| download | bcm5719-llvm-0568e952b6d1dc53f44c8eb5af167fc9d2e0bb34.tar.gz bcm5719-llvm-0568e952b6d1dc53f44c8eb5af167fc9d2e0bb34.zip | |
Add a utility accessor 'has_single_element' for ranges.
This provides an easy way to check if a range has a single element.
PiperOrigin-RevId: 277544647
Diffstat (limited to 'mlir/lib/Dialect/SPIRV/SPIRVOps.cpp')
| -rw-r--r-- | mlir/lib/Dialect/SPIRV/SPIRVOps.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/mlir/lib/Dialect/SPIRV/SPIRVOps.cpp b/mlir/lib/Dialect/SPIRV/SPIRVOps.cpp index 85e22a5c391..c47d4b87027 100644 --- a/mlir/lib/Dialect/SPIRV/SPIRVOps.cpp +++ b/mlir/lib/Dialect/SPIRV/SPIRVOps.cpp @@ -1418,7 +1418,7 @@ static void print(spirv::LoopOp loopOp, OpAsmPrinter &printer) { /// given `dstBlock`. static inline bool hasOneBranchOpTo(Block &srcBlock, Block &dstBlock) { // Check that there is only one op in the `srcBlock`. - if (srcBlock.empty() || std::next(srcBlock.begin()) != srcBlock.end()) + if (!has_single_element(srcBlock)) return false; auto branchOp = dyn_cast<spirv::BranchOp>(srcBlock.back()); |
