Skip to content

Commit 50f1605

Browse files
[CI] Fix AWS CUDA CI in precommit (#9128)
The AWS CUDA testing in pre-commit unwillingly got disabled in #8656. This PR re-enables the AWS CUDA testing in pre-commit.
1 parent 04e349b commit 50f1605

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

devops/scripts/generate_test_matrix.js

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,9 +50,18 @@ module.exports = ({core, process}) => {
5050
} else {
5151
v["env"] = {};
5252
}
53+
54+
// Check for CUDA machines. If available, add them to
55+
// enabledLTSLxConfigs.
56+
var hasCuda = false;
57+
if (Array.isArray(v["runs-on"]))
58+
hasCuda = v["runs-on"].some(e => e.includes("aws-cuda"));
59+
else
60+
hasCuda = v["runs-on"].includes("aws-cuda")
61+
5362
if (v["runs-on"].includes("Windows"))
5463
enabledLTSWnConfigs.push(v);
55-
else if (v["runs-on"].includes("Linux"))
64+
else if (v["runs-on"].includes("Linux") || hasCuda)
5665
enabledLTSLxConfigs.push(v);
5766
else
5867
console.error("runs-on OS is not recognized");

0 commit comments

Comments
 (0)