Skip to content

Commit 9459e64

Browse files
[CI][Bench] Use GPU mask to match the CPU mask
and print produced results in CI, just FYI.
1 parent 95e03b4 commit 9459e64

File tree

1 file changed

+10
-2
lines changed

1 file changed

+10
-2
lines changed

.github/workflows/benchmarks-reusable.yml

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -182,8 +182,8 @@ jobs:
182182

183183
- name: Compute core range
184184
run: |
185-
# Compute the core range for the first NUMA node, skipping the first 4 cores.
186-
# This is to avoid the first cores that the kernel is likely to schedule more work on.
185+
# Compute the core range for the first NUMA node; second node is for UMF jobs.
186+
# Skip the first 4 cores - the kernel is likely to schedule more work on these.
187187
CORES=$(lscpu | awk '
188188
/NUMA node0 CPU|On-line CPU/ {line=$0}
189189
END {
@@ -192,8 +192,12 @@ jobs:
192192
sub(/^0/, "4", b[1])
193193
print b[1]
194194
}')
195+
echo "Selected core: $CORES"
195196
echo "CORES=$CORES" >> $GITHUB_ENV
196197
198+
ZE_AFFINITY_MASK=0
199+
echo "ZE_AFFINITY_MASK=$ZE_AFFINITY_MASK" >> $GITHUB_ENV
200+
197201
- name: Run benchmarks
198202
working-directory: ${{ github.workspace }}/ur-repo/
199203
id: benchmarks
@@ -209,6 +213,10 @@ jobs:
209213
${{ inputs.upload_report && '--output-html' || '' }}
210214
${{ inputs.bench_script_params }}
211215
216+
- name: Print benchmark results
217+
run: |
218+
cat ${{ github.workspace }}/ur-repo/benchmark_results.md
219+
212220
- name: Add comment to PR
213221
uses: actions/github-script@60a0d83039c74a4aee543508d2ffcb1c3799cdea # v7.0.1
214222
if: ${{ always() && inputs.pr_no != 0 }}

0 commit comments

Comments
 (0)