mirrors/kaldifeat
1
0
Fork 0
mirror of https://github.com/csukuangfj/kaldifeat.git synced 2025-08-09 01:52:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

support for torch 2.4.1 (#106)

Browse Source
This commit is contained in:
Fangjun Kuang 2024-09-07 17:32:29 +08:00 committed by GitHub
parent a39a011de9
commit 5f9602afab
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 115 additions and 68 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

145
scripts/github_actions/generate_build_matrix.py
View File

@ -71,27 +71,35 @@ def generate_build_matrix(enable_cuda, for_windows, for_macos, test_only_latest_
# }, # },
"1.7.0": { "1.7.0": {
"python-version": ["3.6", "3.7", "3.8"], "python-version": ["3.6", "3.7", "3.8"],
"cuda": ["10.1", "10.2", "11.0"] "cuda": (
["10.1", "10.2", "11.0"]
if not for_windows if not for_windows
else ["10.1.243", "10.2.89", "11.0.3"], else ["10.1.243", "10.2.89", "11.0.3"]
),
}, },
"1.7.1": { "1.7.1": {
"python-version": ["3.6", "3.7", "3.8", "3.9"], "python-version": ["3.6", "3.7", "3.8", "3.9"],
"cuda": ["10.1", "10.2", "11.0"] "cuda": (
["10.1", "10.2", "11.0"]
if not for_windows if not for_windows
else ["10.1.243", "10.2.89", "11.0.3"], else ["10.1.243", "10.2.89", "11.0.3"]
),
}, },
"1.8.0": { "1.8.0": {
"python-version": ["3.6", "3.7", "3.8", "3.9"], "python-version": ["3.6", "3.7", "3.8", "3.9"],
"cuda": ["10.1", "10.2", "11.1"] "cuda": (
["10.1", "10.2", "11.1"]
if not for_windows if not for_windows
else ["10.1.243", "10.2.89", "11.1.1"], else ["10.1.243", "10.2.89", "11.1.1"]
),
}, },
"1.8.1": { "1.8.1": {
"python-version": ["3.6", "3.7", "3.8", "3.9"], "python-version": ["3.6", "3.7", "3.8", "3.9"],
"cuda": ["10.1", "10.2", "11.1"] "cuda": (
["10.1", "10.2", "11.1"]
if not for_windows if not for_windows
else ["10.1.243", "10.2.89", "11.1.1"], else ["10.1.243", "10.2.89", "11.1.1"]
),
}, },
"1.9.0": { "1.9.0": {
"python-version": ["3.6", "3.7", "3.8", "3.9"], "python-version": ["3.6", "3.7", "3.8", "3.9"],
@ -103,39 +111,45 @@ def generate_build_matrix(enable_cuda, for_windows, for_macos, test_only_latest_
}, },
"1.10.0": { "1.10.0": {
"python-version": ["3.6", "3.7", "3.8", "3.9"], "python-version": ["3.6", "3.7", "3.8", "3.9"],
"cuda": ["10.2", "11.1", "11.3"] "cuda": (
["10.2", "11.1", "11.3"]
if not for_windows if not for_windows
else ["10.2.89", "11.1.1", "11.3.1"], else ["10.2.89", "11.1.1", "11.3.1"]
),
}, },
"1.10.1": { "1.10.1": {
"python-version": ["3.6", "3.7", "3.8", "3.9"], "python-version": ["3.6", "3.7", "3.8", "3.9"],
"cuda": ["10.2", "11.1", "11.3"] "cuda": (
["10.2", "11.1", "11.3"]
if not for_windows if not for_windows
else ["10.2.89", "11.1.1", "11.3.1"], else ["10.2.89", "11.1.1", "11.3.1"]
),
}, },
"1.10.2": { "1.10.2": {
"python-version": ["3.6", "3.7", "3.8", "3.9"], "python-version": ["3.6", "3.7", "3.8", "3.9"],
"cuda": ["10.2", "11.1", "11.3"] "cuda": (
["10.2", "11.1", "11.3"]
if not for_windows if not for_windows
else ["10.2.89", "11.1.1", "11.3.1"], else ["10.2.89", "11.1.1", "11.3.1"]
),
}, },
"1.11.0": { "1.11.0": {
"python-version": ["3.7", "3.8", "3.9", "3.10"], "python-version": ["3.7", "3.8", "3.9", "3.10"],
"cuda": ["10.2", "11.3", "11.5"] "cuda": (
if not for_windows ["10.2", "11.3", "11.5"] if not for_windows else ["11.3.1", "11.5.2"]
else ["11.3.1", "11.5.2"], ),
}, },
"1.12.0": { "1.12.0": {
"python-version": ["3.7", "3.8", "3.9", "3.10"], "python-version": ["3.7", "3.8", "3.9", "3.10"],
"cuda": ["10.2", "11.3", "11.6"] "cuda": (
if not for_windows ["10.2", "11.3", "11.6"] if not for_windows else ["11.3.1", "11.6.2"]
else ["11.3.1", "11.6.2"], ),
}, },
"1.12.1": { "1.12.1": {
"python-version": ["3.7", "3.8", "3.9", "3.10"], "python-version": ["3.7", "3.8", "3.9", "3.10"],
"cuda": ["10.2", "11.3", "11.6"] "cuda": (
if not for_windows ["10.2", "11.3", "11.6"] if not for_windows else ["11.3.1", "11.6.2"]
else ["11.3.1", "11.6.2"], ),
}, },
"1.13.0": { "1.13.0": {
"python-version": ["3.7", "3.8", "3.9", "3.10", "3.11"], "python-version": ["3.7", "3.8", "3.9", "3.10", "3.11"],
@ -143,80 +157,112 @@ def generate_build_matrix(enable_cuda, for_windows, for_macos, test_only_latest_
}, },
"1.13.1": { "1.13.1": {
"python-version": ["3.7", "3.8", "3.9", "3.10", "3.11"], "python-version": ["3.7", "3.8", "3.9", "3.10", "3.11"],
"cuda": ["11.6", "11.7"] # default 11.7 "cuda": (
["11.6", "11.7"] # default 11.7
if not for_windows if not for_windows
else ["11.6.2", "11.7.1"], else ["11.6.2", "11.7.1"]
),
}, },
"2.0.0": { "2.0.0": {
"python-version": ["3.8", "3.9", "3.10", "3.11"], "python-version": ["3.8", "3.9", "3.10", "3.11"],
"cuda": ["11.7", "11.8"] # default 11.7 "cuda": (
["11.7", "11.8"] # default 11.7
if not for_windows if not for_windows
else ["11.7.1", "11.8.0"], else ["11.7.1", "11.8.0"]
),
}, },
"2.0.1": { "2.0.1": {
"python-version": ["3.8", "3.9", "3.10", "3.11"], "python-version": ["3.8", "3.9", "3.10", "3.11"],
"cuda": ["11.7", "11.8"] # default 11.7 "cuda": (
["11.7", "11.8"] # default 11.7
if not for_windows if not for_windows
else ["11.7.1", "11.8.0"], else ["11.7.1", "11.8.0"]
),
}, },
"2.1.0": { "2.1.0": {
"python-version": ["3.8", "3.9", "3.10", "3.11"], "python-version": ["3.8", "3.9", "3.10", "3.11"],
"cuda": ["11.8", "12.1"] # default 12.1 "cuda": (
["11.8", "12.1"] # default 12.1
if not for_windows if not for_windows
else ["11.8.0", "12.1.0"], else ["11.8.0", "12.1.0"]
),
}, },
"2.1.1": { "2.1.1": {
"python-version": ["3.8", "3.9", "3.10", "3.11"], "python-version": ["3.8", "3.9", "3.10", "3.11"],
"cuda": ["11.8", "12.1"] # default 12.1 "cuda": (
["11.8", "12.1"] # default 12.1
if not for_windows if not for_windows
else ["11.8.0", "12.1.0"], else ["11.8.0", "12.1.0"]
),
}, },
"2.1.2": { "2.1.2": {
"python-version": ["3.8", "3.9", "3.10", "3.11"], "python-version": ["3.8", "3.9", "3.10", "3.11"],
"cuda": ["11.8", "12.1"] # default 12.1 "cuda": (
["11.8", "12.1"] # default 12.1
if not for_windows if not for_windows
else ["11.8.0", "12.1.0"], else ["11.8.0", "12.1.0"]
),
}, },
"2.2.0": { "2.2.0": {
"python-version": ["3.8", "3.9", "3.10", "3.11", "3.12"], "python-version": ["3.8", "3.9", "3.10", "3.11", "3.12"],
"cuda": ["11.8", "12.1"] # default 12.1 "cuda": (
["11.8", "12.1"] # default 12.1
if not for_windows if not for_windows
else ["11.8.0", "12.1.0"], else ["11.8.0", "12.1.0"]
),
}, },
"2.2.1": { "2.2.1": {
"python-version": ["3.8", "3.9", "3.10", "3.11", "3.12"], "python-version": ["3.8", "3.9", "3.10", "3.11", "3.12"],
"cuda": ["11.8", "12.1"] # default 12.1 "cuda": (
["11.8", "12.1"] # default 12.1
if not for_windows if not for_windows
else ["11.8.0", "12.1.0"], else ["11.8.0", "12.1.0"]
),
}, },
"2.2.2": { "2.2.2": {
"python-version": ["3.8", "3.9", "3.10", "3.11", "3.12"], "python-version": ["3.8", "3.9", "3.10", "3.11", "3.12"],
"cuda": ["11.8", "12.1"] # default 12.1 "cuda": (
["11.8", "12.1"] # default 12.1
if not for_windows if not for_windows
else ["11.8.0", "12.1.0"], else ["11.8.0", "12.1.0"]
),
}, },
"2.3.0": { "2.3.0": {
"python-version": ["3.8", "3.9", "3.10", "3.11", "3.12"], "python-version": ["3.8", "3.9", "3.10", "3.11", "3.12"],
"cuda": ["11.8", "12.1"] # default 12.1 "cuda": (
["11.8", "12.1"] # default 12.1
if not for_windows if not for_windows
else ["11.8.0", "12.1.0"], else ["11.8.0", "12.1.0"]
),
}, },
"2.3.1": { "2.3.1": {
"python-version": ["3.8", "3.9", "3.10", "3.11", "3.12"], "python-version": ["3.8", "3.9", "3.10", "3.11", "3.12"],
"cuda": ["11.8", "12.1"] # default 12.1 "cuda": (
["11.8", "12.1"] # default 12.1
if not for_windows if not for_windows
else ["11.8.0", "12.1.0"], else ["11.8.0", "12.1.0"]
),
}, },
"2.4.0": { "2.4.0": {
"python-version": ["3.8", "3.9", "3.10", "3.11", "3.12"], "python-version": ["3.8", "3.9", "3.10", "3.11", "3.12"],
"cuda": ["11.8", "12.1", "12.4"] # default 12.1 "cuda": (
["11.8", "12.1", "12.4"] # default 12.1
if not for_windows if not for_windows
else ["11.8.0", "12.1.0", "12.4.0"], else ["11.8.0", "12.1.0", "12.4.0"]
),
},
"2.4.1": {
"python-version": ["3.8", "3.9", "3.10", "3.11", "3.12"],
"cuda": (
["11.8", "12.1", "12.4"] # default 12.1
if not for_windows
else ["11.8.0", "12.1.0", "12.4.0"]
),
}, },
# https://github.com/Jimver/cuda-toolkit/blob/master/src/links/windows-links.ts # https://github.com/Jimver/cuda-toolkit/blob/master/src/links/windows-links.ts
} }
if test_only_latest_torch: if test_only_latest_torch:
latest = "2.4.0" latest = "2.4.1"
matrix = {latest: matrix[latest]} matrix = {latest: matrix[latest]}
if for_windows or for_macos: if for_windows or for_macos:
@ -263,7 +309,8 @@ def generate_build_matrix(enable_cuda, for_windows, for_macos, test_only_latest_
{ {
"torch": torch, "torch": torch,
"python-version": p, "python-version": p,
"image": "pytorch/manylinux-builder:cpu-2.4", # "image": "pytorch/manylinux-builder:cpu-2.4",
"image": "pytorch/manylinux-builder:cpu-27677ead7c8293c299a885ae2c474bf445e653a5",
} }
) )
elif version_ge(torch, "2.2.0"): elif version_ge(torch, "2.2.0"):