xla icon indicating copy to clipboard operation
xla copied to clipboard

[ROCm] Make multi gpu tests exclusive if executed locally

Open alekstheod opened this issue 1 month ago โ€ข 1 comments

๐Ÿ“ Summary of Changes Make multigpu tests being executed exclusively.

๐ŸŽฏ Justification Multigpu tests have to run exclusively to not clash the utilization of gpus in parallel. This PR marks multigpu tests as such by adding 'multi_gpu' tag, and in case of rocm make them run exclusively if executed locally.

๐Ÿš€ Kind of Contribution Please remove what does not apply: โ™ป๏ธ Cleanup

๐Ÿ“Š Benchmark (for Performance Improvements) Not relevant, no logic change

๐Ÿงช Unit Tests: Not relevant, no logic change

๐Ÿงช Execution Tests: Not relevant, no logic change

alekstheod avatar Nov 13 '25 15:11 alekstheod

@beckerhe had a comment on that. It seems that xla_cc_tests are not really gpu tests but cpu only tests. I will have to remove the tag and adjust accordingly!

alekstheod avatar Nov 13 '25 18:11 alekstheod

@xla-rotation please check this PR

alekstheod avatar Nov 18 '25 11:11 alekstheod