flat

package

v1.0.1 Latest Latest Go to latest Published: Feb 11, 2026 License: MIT Imports: 5 Imported by: 0

Details

This section is empty.

This section is empty.

This section is empty.

type FlatIndex struct {
	// contains filtered or unexported fields
}

FlatIndex is a simple flat index for exact search

func NewFlatIndex(capacity int) *FlatIndex

NewFlatIndex creates a new flat index

func (idx *FlatIndex) Add(vec simd.Vec512, scale float32, id int)

Add adds a vector to the index

func (idx *FlatIndex) AddBatch(vecs []simd.Vec512, scales []float32, ids []int)

AddBatch adds multiple vectors at once

func (idx *FlatIndex) SearchTopK(query *simd.Vec512, K int) []SearchResult

SearchTopK performs exact search and returns top K results

func (idx *FlatIndex) SearchTopKBatch(queries []*simd.Vec512, K int) [][]SearchResult

SearchTopKBatch processes multiple queries in batch for better cache utilization

func (idx *FlatIndex) SearchTopKParallel(query *simd.Vec512, K int) []SearchResult

SearchTopKParallel performs parallel exact search for better throughput

func (idx *FlatIndex) SearchTopKTiled(query *simd.Vec512, K int) []SearchResult

Optimized version with cache tiling

func (idx *FlatIndex) Size() int

Size returns the number of vectors in the index

type SearchResult struct {
	ID    int
	Score int32
}

SearchResult represents a search result with ID and score