maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

Fix raw buffer size when creating a DenseElementsAttr from an array of attributes. 

PiperOrigin-RevId: 229973134
This commit is contained in:
River Riddle 2019-01-18 12:23:47 -08:00 committed by jpienaar
parent 1e484b5ef4
commit b04c9a47ca
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mlir/lib/IR/MLIRContext.cpp
View File

@ -1078,6 +1078,8 @@ DenseElementsAttr DenseElementsAttr::get(VectorOrTensorType type,
ArrayRef<Attribute> values) {
assert(type.getElementType().isIntOrFloat() &&
"expected int or float element type");
assert(values.size() == type.getNumElements() &&
"expected 'values' to contain the same number of elements as 'type'");
// FIXME: using 64 bits for BF16 because it is currently stored with double
// semantics.
@ -1085,7 +1087,7 @@ DenseElementsAttr DenseElementsAttr::get(VectorOrTensorType type,
size_t bitWidth = eltType.isBF16() ? 64 : eltType.getIntOrFloatBitWidth();
// Compress the attribute values into a character buffer.
SmallVector<char, 8> data(type.getSizeInBits() * 8L);
SmallVector<char, 8> data(APInt::getNumWords(bitWidth * values.size()) * 8L);
for (unsigned i = 0, e = values.size(); i < e; ++i) {
unsigned bitPos = i * bitWidth;