mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-08 09:32:20 +00:00
110 lines
3.1 KiB
Python
Executable File
110 lines
3.1 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# Copyright 2023 Xiaomi Corp. (authors: Fangjun Kuang)
|
|
|
|
|
|
import json
|
|
|
|
|
|
def version_gt(a, b):
|
|
a_major, a_minor = list(map(int, a.split(".")))[:2]
|
|
b_major, b_minor = list(map(int, b.split(".")))[:2]
|
|
if a_major > b_major:
|
|
return True
|
|
|
|
if a_major == b_major and a_minor > b_minor:
|
|
return True
|
|
|
|
return False
|
|
|
|
|
|
def version_ge(a, b):
|
|
a_major, a_minor = list(map(int, a.split(".")))[:2]
|
|
b_major, b_minor = list(map(int, b.split(".")))[:2]
|
|
if a_major > b_major:
|
|
return True
|
|
|
|
if a_major == b_major and a_minor >= b_minor:
|
|
return True
|
|
|
|
return False
|
|
|
|
|
|
def get_torchaudio_version(torch_version):
|
|
if torch_version == "1.13.0":
|
|
return "0.13.0"
|
|
elif torch_version == "1.13.1":
|
|
return "0.13.1"
|
|
elif torch_version == "2.0.0":
|
|
return "2.0.1"
|
|
elif torch_version == "2.0.1":
|
|
return "2.0.2"
|
|
else:
|
|
return torch_version
|
|
|
|
|
|
def get_matrix():
|
|
k2_version = "1.24.4.dev20240223"
|
|
kaldifeat_version = "1.25.4.dev20240223"
|
|
version = "20240905"
|
|
python_version = ["3.8", "3.9", "3.10", "3.11", "3.12"]
|
|
torch_version = []
|
|
torch_version += ["1.13.0", "1.13.1"]
|
|
torch_version += ["2.0.0", "2.0.1"]
|
|
torch_version += ["2.1.0", "2.1.1", "2.1.2"]
|
|
torch_version += ["2.2.0", "2.2.1", "2.2.2"]
|
|
torch_version += ["2.3.0", "2.3.1"]
|
|
torch_version += ["2.4.0"]
|
|
torch_version += ["2.4.1"]
|
|
|
|
matrix = []
|
|
for p in python_version:
|
|
for t in torch_version:
|
|
# torchaudio <= 1.13.x supports only python <= 3.10
|
|
|
|
if version_gt(p, "3.10") and not version_gt(t, "2.0"):
|
|
continue
|
|
|
|
# only torch>=2.2.0 supports python 3.12
|
|
if version_gt(p, "3.11") and not version_gt(t, "2.1"):
|
|
continue
|
|
|
|
k2_version_2 = k2_version
|
|
kaldifeat_version_2 = kaldifeat_version
|
|
|
|
if t == "2.2.2":
|
|
k2_version_2 = "1.24.4.dev20240328"
|
|
kaldifeat_version_2 = "1.25.4.dev20240329"
|
|
elif t == "2.3.0":
|
|
k2_version_2 = "1.24.4.dev20240425"
|
|
kaldifeat_version_2 = "1.25.4.dev20240425"
|
|
elif t == "2.3.1":
|
|
k2_version_2 = "1.24.4.dev20240606"
|
|
kaldifeat_version_2 = "1.25.4.dev20240606"
|
|
elif t == "2.4.0":
|
|
k2_version_2 = "1.24.4.dev20240725"
|
|
kaldifeat_version_2 = "1.25.4.dev20240725"
|
|
elif t == "2.4.1":
|
|
k2_version_2 = "1.24.4.dev20240905"
|
|
kaldifeat_version_2 = "1.25.4.dev20240905"
|
|
|
|
matrix.append(
|
|
{
|
|
"k2-version": k2_version_2,
|
|
"kaldifeat-version": kaldifeat_version_2,
|
|
"version": version,
|
|
"python-version": p,
|
|
"torch-version": t,
|
|
"torchaudio-version": get_torchaudio_version(t),
|
|
}
|
|
)
|
|
return matrix
|
|
|
|
|
|
def main():
|
|
matrix = get_matrix()
|
|
print(json.dumps({"include": matrix}))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|