forked from dusty-nv/jetson-containers
-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.py
36 lines (26 loc) · 950 Bytes
/
config.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
def faiss(version, branch=None, requires=None, default=False):
pkg = package.copy()
pkg['name'] = f'faiss:{version}'
if len(version.split('.')) < 3:
version = version + '.0'
if not branch:
branch = 'v' + version
pkg['build_args'] = {
'FAISS_VERSION': version,
'FAISS_BRANCH': branch,
}
if requires:
pkg['requires'] = requires
builder = pkg.copy()
builder['name'] = builder['name'] + '-builder'
builder['build_args'] = {**builder['build_args'], 'FORCE_BUILD': 'on'}
if default:
pkg['alias'] = 'faiss'
builder['alias'] = 'faiss:builder'
return pkg, builder
package = [
faiss('1.7.3'),
faiss('1.7.4', default=True),
#faiss('v1.8.0'), # encounters type_info build error sometime after be12427 (12/12/2023)
#faiss('be12427', default=True), # known good build on JP5/JP6 from 12/12/2023
]