make env var paths independent of ROCm version

[JehandadKhan]

Motivation

A mismatch in LLVM_PATH or ROCM_PATH based on the minor version breaks logic in XLA where we look for tools such as ld.lld

Technical Details

This PR fixes such issues by using the symlink path instead which is stable across versions.

Test Plan

Once the docker is built a simple code snippet such as

import jax.numpy as jnp
print(jnp.ones(5))

Should work successfully, if the path is not set correctly, this will fail.

Test Result

Submission Checklist

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

[Copilot]

Copilot encountered an error and was unable to review this pull request. You can try again by re-requesting a review.