HNSWFlat index with JensenShannon metric brings program stuck

[qwevdb]

Summary

When using HNSWFlat index with JensenShannon metric and processed data > 1.0, the program will get stuck.

Platform

OS: Ubuntu 24.04 LTS

Faiss version: faiss-gpu 1.8.0

Installed from: Anaconda

Faiss compilation options:

Running on:

• [✔] CPU
• GPU

Interface:

• C++
• [✔] Python

Reproduction instructions

When using HNSWFlat index with JensenShannon metric and handling data greater than 1.0, the Python script will get stuck.
However, the Python script runs normally, when handling data between 0 and 1.0 .

import faiss
import numpy as np

d = 2
metric_type = faiss.METRIC_JensenShannon
M = 10
index = faiss.IndexHNSWFlat(d, M, metric_type)

np.random.seed(0)
insert_data = np.random.randint(0, 10, (10, d))  # stuck
# insert_data = np.random.randn(10, d)           # stuck
# insert_data = np.random.random((10, d))        # not stuck

index.train(insert_data)
index.add(insert_data)

query_data = np.random.randint(0, 10, (1, d))
# query_data = np.random.randn(1, d)         
# query_data = np.random.random((1, d))

k = 5

#program stuck
D, I = index.search(query_data, k)
print(D)
print(I)