liuyu/pgvector
1
0
Fork 0
mirror of https://github.com/pgvector/pgvector.git synced 2026-07-14 16:46:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

Updated style [skip ci]

Browse Source
This commit is contained in:
Andrew Kane
2026-04-14 16:21:32 -07:00
parent 5c9a97af3b
commit a7551a61ca
1 changed files with 8 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/ivfbuild.c
View File

@@ -63,8 +63,8 @@ AddSample(Datum *values, IvfflatBuildState * buildstate)
Datum value = PointerGetDatum(PG_DETOAST_DATUM(values[0])); Datum value = PointerGetDatum(PG_DETOAST_DATUM(values[0]));
/* /*
* Check with KMEANS_NORM_PROC that the value can be normalized * Check with KMEANS_NORM_PROC that the value can be normalized since
* since spherical distance function expects unit vectors * spherical distance function expects unit vectors
*/ */
if (buildstate->kmeansnormprocinfo != NULL) if (buildstate->kmeansnormprocinfo != NULL)
{ {
@@ -150,18 +150,17 @@ SampleRows(IvfflatBuildState * buildstate)
} }
/* Normalize if needed */ /* Normalize if needed */
if (buildstate->kmeansnormprocinfo) if (buildstate->kmeansnormprocinfo != NULL)
{ {
VectorArray samples = buildstate->samples; VectorArray samples = buildstate->samples;
Datum value;
Datum nvalue;
for (int i = 0; i < samples->length; i++) for (int i = 0; i < samples->length; i++)
{ {
value = PointerGetDatum(VectorArrayGet(samples, i)); Datum value = PointerGetDatum(VectorArrayGet(samples, i));
nvalue = IvfflatNormValue(buildstate->typeInfo, buildstate->collation, value); Datum normValue = IvfflatNormValue(buildstate->typeInfo, buildstate->collation, value);
VectorArraySet(samples, i, DatumGetPointer(nvalue));
pfree(DatumGetPointer(nvalue)); VectorArraySet(samples, i, DatumGetPointer(normValue));
pfree(DatumGetPointer(normValue));
} }
} }
} }