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

Added support function for l2_normalize [skip ci]

Browse Source
This commit is contained in:
Andrew Kane
2024-04-22 18:36:47 -07:00
parent 2b77005610
commit f14c21748b
8 changed files with 22 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/hnswbuild.c
View File

@@ -496,7 +496,7 @@ InsertTuple(Relation index, Datum *values, bool *isnull, ItemPointer heaptid, Hn
if (!HnswCheckNorm(buildstate->normprocinfo, buildstate->collation, value))
return false;
value = HnswNormValue(value, buildstate->type);
value = HnswNormValue(buildstate->normalizeprocinfo, buildstate->collation, value);
}
/* Get datum size */
@@ -725,6 +725,7 @@ InitBuildState(HnswBuildState * buildstate, Relation heap, Relation index, Index
/* Get support functions */
buildstate->procinfo = index_getprocinfo(index, 1, HNSW_DISTANCE_PROC);
buildstate->normprocinfo = HnswOptionalProcInfo(index, HNSW_NORM_PROC);
buildstate->normalizeprocinfo = HnswOptionalProcInfo(index, HNSW_NORMALIZE_PROC);
buildstate->collation = index->rd_indcollation[0];
InitGraph(&buildstate->graphData, NULL, maintenance_work_mem * 1024L);