xla
xla copied to clipboard
[ROCm] Make rbe amdgpu pools configurable
๐ 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