liuyu/pgvector
1
0
Fork 0
mirror of https://github.com/pgvector/pgvector.git synced 2026-07-01 10:11:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

Ensure items are always aligned

Browse Source
This commit is contained in:
Andrew Kane
2024-04-17 00:17:40 -07:00
parent cf570810eb
commit 576a37e975
2 changed files with 6 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/ivfbuild.c
View File

@@ -337,20 +337,14 @@ GetMaxDimensions(IvfflatType type)
static Size
GetItemSize(IvfflatType type, int dimensions)
{
Size size;
if (type == IVFFLAT_TYPE_VECTOR)
size = VECTOR_SIZE(dimensions);
return VECTOR_SIZE(dimensions);
else if (type == IVFFLAT_TYPE_HALFVEC)
size = HALFVEC_SIZE(dimensions);
return HALFVEC_SIZE(dimensions);
else if (type == IVFFLAT_TYPE_BIT)
size = VARBITTOTALLEN(dimensions);
return VARBITTOTALLEN(dimensions);
else
elog(ERROR, "Unsupported type");
/* Ensure items are aligned to prevent UB */
/* Only need 4 byte alignment, but prefer MAXALIGN */
return MAXALIGN(size);
}
/*

3
src/ivfutils.c
View File

@@ -18,6 +18,9 @@ VectorArrayInit(int maxlen, int dimensions, Size itemsize)
{
VectorArray res = palloc(sizeof(VectorArrayData));
/* Ensure items are aligned to prevent UB */
itemsize = MAXALIGN(itemsize);
res->length = 0;
res->maxlen = maxlen;
res->dim = dimensions;