[Mlir-commits] [mlir] 83274a0 - [mlir] Add SmallVector sizes

Tres Popp llvmlistbot at llvm.org
Mon Dec 21 15:48:52 PST 2020


Author: Tres Popp
Date: 2020-12-22T00:48:41+01:00
New Revision: 83274a0773f6a20abdc848b448009e0195c42166

URL: https://github.com/llvm/llvm-project/commit/83274a0773f6a20abdc848b448009e0195c42166
DIFF: https://github.com/llvm/llvm-project/commit/83274a0773f6a20abdc848b448009e0195c42166.diff

LOG: [mlir] Add SmallVector sizes

This is a temporary fix until figuring out how to correct the forward
declare in mlir/include/mlir/Support/LLVM.h

Differential Revision: https://reviews.llvm.org/D93666

Added: 
    

Modified: 
    mlir/lib/Dialect/LLVMIR/IR/LLVMDialect.cpp

Removed: 
    


################################################################################
diff  --git a/mlir/lib/Dialect/LLVMIR/IR/LLVMDialect.cpp b/mlir/lib/Dialect/LLVMIR/IR/LLVMDialect.cpp
index 7b1300da1783..09c662c74477 100644
--- a/mlir/lib/Dialect/LLVMIR/IR/LLVMDialect.cpp
+++ b/mlir/lib/Dialect/LLVMIR/IR/LLVMDialect.cpp
@@ -198,8 +198,8 @@ void SwitchOp::build(OpBuilder &builder, OperationState &result, Value value,
                      ArrayRef<int32_t> caseValues, BlockRange caseDestinations,
                      ArrayRef<ValueRange> caseOperands,
                      ArrayRef<int32_t> branchWeights) {
-  SmallVector<Value> flattenedCaseOperands;
-  SmallVector<int32_t> caseOperandOffsets;
+  SmallVector<Value, 8> flattenedCaseOperands;
+  SmallVector<int32_t, 8> caseOperandOffsets;
   int32_t offset = 0;
   for (ValueRange operands : caseOperands) {
     flattenedCaseOperands.append(operands.begin(), operands.end());
@@ -230,8 +230,8 @@ parseSwitchOpCases(OpAsmParser &parser, ElementsAttr &caseValues,
                    SmallVectorImpl<OpAsmParser::OperandType> &caseOperands,
                    SmallVectorImpl<Type> &caseOperandTypes,
                    ElementsAttr &caseOperandOffsets) {
-  SmallVector<int32_t> values;
-  SmallVector<int32_t> offsets;
+  SmallVector<int32_t, 8> values;
+  SmallVector<int32_t, 8> offsets;
   int32_t value, offset = 0;
   do {
     OptionalParseResult integerParseResult = parser.parseOptionalInteger(value);
@@ -243,7 +243,7 @@ parseSwitchOpCases(OpAsmParser &parser, ElementsAttr &caseValues,
     values.push_back(value);
 
     Block *destination;
-    SmallVector<OpAsmParser::OperandType> operands;
+    SmallVector<OpAsmParser::OperandType, 8> operands;
     if (parser.parseColon() || parser.parseSuccessor(destination))
       return failure();
     if (!parser.parseOptionalLParen()) {


        


More information about the Mlir-commits mailing list