Merge pull request #170 from vajain-rhods/fix_rocm_build

Fix vLLM ROCm build

Author: riprasad

Dockerfile.rocm.ubi

@@ -210,7 +210,7 @@ RUN --mount=type=bind,from=build_amdsmi,src=/install,target=/install/amdsmi/ \
     --mount=type=bind,from=build_vllm,src=/workspace/dist,target=/install/vllm/ \
     --mount=type=cache,target=/root/.cache/pip \
     --mount=type=cache,target=/root/.cache/uv \
-    export version="$(awk -F. '{print $1"."$2}' <<< $ROCM_VERSION)" && \
+    export version="$(awk -F. '{print $1"."$2}' <<< ${ROCM_VERSION})" && \
     uv pip install \
         --index-strategy=unsafe-best-match \
         --extra-index-url "https://download.pytorch.org/whl/nightly/rocm${version}" \

requirements/rocm.txt

@@ -4,8 +4,6 @@
 numba == 0.60.0; python_version == '3.9' # v0.61 doesn't support Python 3.9. Required for N-gram speculative decoding
 numba == 0.61.2; python_version > '3.9'
 
-numba == 0.60.0 # v0.61 doesn't support Python 3.9. Required for N-gram speculative decoding.
-
 # Dependencies for AMD GPUs
 awscli
 boto3