mirrors/kaldifeat
1
0
Fork 0
mirror of https://github.com/csukuangfj/kaldifeat.git synced 2025-08-22 16:26:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

small fixes

Browse Source
This commit is contained in:
Fangjun Kuang 2023-07-22 10:32:08 +08:00
parent e57dc5cf3b
commit bb46dd21b5
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
scripts/github_actions/generate_build_matrix.py
View File

@ -128,9 +128,17 @@ def generate_build_matrix(enable_cuda, for_windows, test_only_latest_torch):
"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
}, },
"2.0.0": {
"python-version": ["3.8", "3.9", "3.10", "3.11"],
"cuda": ["11.7", "11.8"], # default 11.7
},
"2.0.1": {
"python-version": ["3.8", "3.9", "3.10", "3.11"],
"cuda": ["11.7", "11.8"], # default 11.7
},
} }
if test_only_latest_torch: if test_only_latest_torch:
latest = "1.13.0" latest = "2.0.1"
matrix = {latest: matrix[latest]} matrix = {latest: matrix[latest]}
ans = [] ans = []
@ -143,6 +151,8 @@ def generate_build_matrix(enable_cuda, for_windows, test_only_latest_torch):
ans.append({"torch": torch, "python-version": p, "cuda": c}) ans.append({"torch": torch, "python-version": p, "cuda": c})
else: else:
for p in python_versions: for p in python_versions:
if for_windows:
p = "cp" + "".join(p.split("."))
ans.append({"torch": torch, "python-version": p}) ans.append({"torch": torch, "python-version": p})
print(json.dumps({"include": ans})) print(json.dumps({"include": ans}))