Update Dockerfile.rocm_ci

[clintg6]

Summary

This PR updates the Dockerfile to ensure the Micromamba environment is automatically activated when the Docker container is started, without requiring the user to manually run activation commands.

Changes

• Appended Micromamba shell hook and environment activation to ~/.bashrc.
  • This ensures the environment is active in all interactive shell sessions (e.g., docker run -it).
• Keeps existing build process and environment creation logic unchanged.

Why?

Previously, users needed to manually activate the Micromamba environment after starting the container:

This update streamlines the user experience by making the environment ready-to-use immediately after container startup.

Test

You can test this by building and running the container interactively:

docker build -f docker/Dockerfile.rocm_ci --target flashinfer_base -t flashinfer-rocm . 2>&1 | tee docker_build.log

and running

docker run -it --network=host --group-add=video \
           --privileged --ipc=host --cap-add=SYS_PTRACE \
           --security-opt seccomp=unconfined --device /dev/kfd \
           --device /dev/dri flashinfer-rocm

Then inside the container, run:

pip show flashinfer

[Copilot]

Pull Request Overview

This PR aims to auto-activate the Micromamba environment in interactive shells by appending initialization and activation commands to the container’s ~/.bashrc.

• Append Micromamba shell hook and environment activation to ~/.bashrc
• Keep existing build and environment creation logic unchanged

---

_{Tip: Customize your code reviews with copilot-instructions.md. Create the file or learn how to get started.}

[diptorupd]

@clintg6 I am fine with these changes. Let @rtmadduri have a look once and I will merge it.