xla icon indicating copy to clipboard operation
xla copied to clipboard

[ROCm] Make rbe amdgpu pools configurable

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

๐Ÿ“ Summary of Changes Making ROCm rbe amdgpu pool name for tests configurable. Preparation for rbe with different gpu archs

๐ŸŽฏ Justification We pass the environment variable that sets the rbe gpu pool name set dynamically. This is needed to execute some builds on a certain gpu arch rbe pool. Especially relevant for hermetic builds where the rocm provided supports a single archictecture.

๐Ÿš€ Kind of Contribution Please remove what does not apply: โœจ New Feature

๐Ÿ“Š Benchmark (for Performance Improvements) Not relevant, rbe setup

๐Ÿงช Unit Tests: Not relevant, rbe setup

๐Ÿงช Execution Tests: Not relevant, rbe setup

alekstheod avatar Nov 13 '25 09:11 alekstheod