mirror of
https://github.com/facebookresearch/faiss.git
synced 2025-06-03 21:54:02 +08:00
eff0898a13
Summary: Pull Request resolved: https://github.com/facebookresearch/faiss/pull/3966 This actually enables the linting. Manual changes: - tools/arcanist/lint/fbsource-licenselint-config.toml - tools/arcanist/lint/fbsource-lint-engine.toml Automated changes: `arc lint --apply-patches --take LICENSELINT --paths-cmd 'hg files faiss'` Reviewed By: asadoughi Differential Revision: D64484165 fbshipit-source-id: 4f2f6e953c94ef6ebfea8a5ae035ccfbea65ed04
36 lines
783 B
Python
36 lines
783 B
Python
#! /usr/bin/env python3
|
|
# Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
#
|
|
# This source code is licensed under the MIT license found in the
|
|
# LICENSE file in the root directory of this source tree.
|
|
|
|
"""small test script to benchmark the SIMD implementation of the
|
|
distance computations for the additional metrics. Call eg. with L1 to
|
|
get L1 distance computations.
|
|
"""
|
|
|
|
import faiss
|
|
|
|
import sys
|
|
import time
|
|
|
|
d = 64
|
|
nq = 4096
|
|
nb = 16384
|
|
|
|
print("sample")
|
|
|
|
xq = faiss.randn((nq, d), 123)
|
|
xb = faiss.randn((nb, d), 123)
|
|
|
|
mt_name = "L2" if len(sys.argv) < 2 else sys.argv[1]
|
|
|
|
mt = getattr(faiss, "METRIC_" + mt_name)
|
|
|
|
print("distances")
|
|
t0 = time.time()
|
|
dis = faiss.pairwise_distances(xq, xb, mt)
|
|
t1 = time.time()
|
|
|
|
print("nq=%d nb=%d d=%d %s: %.3f s" % (nq, nb, d, mt_name, t1 - t0))
|