Fix: Add vim-common package to manylinux Docker image for XLA build

[Ruturaj4]

Motivation

The JAX ROCm plugin wheel build fails with the following error:

ERROR: /root/.cache/bazel/_bazel_root/.../external/xla/xla/backends/gpu/target_config/BUILD:25:12: 
Executing genrule @@xla//xla/backends/gpu/target_config:embed_gpu_specs_gen failed: (Exit 127)

/bin/bash: line 35: xxd: command not found

Target //pjrt/tools:build_gpu_plugin_wheel failed to build

The XLA build process includes a genrule (@@xla//xla/backends/gpu/target_config:embed_gpu_specs_gen) that embeds GPU specification files into C++ code using the xxd utility. The xxd command is used to convert binary/text files into C-style hex arrays:

xxd -i "${src}" | sed -e "s/^unsigned char [^[]*/static const unsigned char ${VAR_NAME}/" ...

The xxd utility is not installed in the manylinux builder Docker image, causing the build to fail.

Solution

Add the vim-common package to the manylinux Dockerfile. On RHEL/AlmaLinux-based systems (which manylinux_2_28 is based on), xxd is provided by the vim-common package.

Submission Checklist

• Look over the contributing guidelines at https://github.com/ROCm/ROCm/blob/develop/CONTRIBUTING.md#pull-requests.