Skip to content

Commit b175045

Browse files
[CI] Increases the number of lit workers based on the GPU arch. (#1919)
* [CI] Increases the number of lit workers based on the GPU arch. - For MFMA architectures (gfx908, gfx90a, gfx942, gfx950), the worker count is increased to 64. - For Navi architectures, the worker count is increased to 16. The goal is to evaluate how CI machines handle more threads, the tests executions remain stable under these settings. This change is an initial step to experiment with performance and stability improvements during CI runs. * Limit LIT worker count for unstable CI architectures. - Reduce LIT worker count to 20 for gfx90a/gfx908 due to CI instability and test timeouts. - Set worker count to 8 for Navi3x/Navi4x due to instability on some navi CI machines. * Temporarily disabled data-after-header validation from tuning file check.
1 parent 35177d6 commit b175045

File tree

2 files changed

+10
-7
lines changed

2 files changed

+10
-7
lines changed

mlir/utils/jenkins/Jenkinsfile

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -355,8 +355,10 @@ def rebootNode() {
355355
int setLitWorkerCount() {
356356
int limit_lit_workers = 8
357357
def gpu_arch = get_gpu_architecture()
358-
if (gpu_arch.contains('gfx942') || gpu_arch.contains('gfx950')) {
359-
limit_lit_workers = 40
358+
if (gpu_arch.contains('gfx908') || gpu_arch.contains('gfx90a')) {
359+
limit_lit_workers = 20
360+
} else if (gpu_arch.contains('gfx942') || gpu_arch.contains('gfx950')) {
361+
limit_lit_workers = 64
360362
}
361363
return limit_lit_workers
362364
}

mlir/utils/tuna/tuna-script.sh

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -59,12 +59,13 @@ validate_tuning_file() {
5959
echo "ERROR: $f has duplicate header lines next to each other!"
6060
exit 1
6161
fi
62+
# NOTE: Temporarily disabled the check for data lines after the header.
6263
# Existing checks
63-
data_line=$(awk 'NR>1 && $0 !~ /^\s*$/ && $0 != header {print; exit}' header="$header" "$f")
64-
if [ -z "$data_line" ]; then
65-
echo "ERROR: $f has no data after header!"
66-
exit 1
67-
fi
64+
#data_line=$(awk 'NR>1 && $0 !~ /^\s*$/ && $0 != header {print; exit}' header="$header" "$f")
65+
#if [ -z "$data_line" ]; then
66+
# echo "ERROR: $f has no data after header!"
67+
# exit 1
68+
#fi
6869
}
6970

7071
function tuna_run

0 commit comments

Comments
 (0)