From 714e9abe52ae9c267bcabbe8c9d854167e289e21 Mon Sep 17 00:00:00 2001
From: pramenku <7664080+pramenku@users.noreply.github.com>
Date: Fri, 13 Dec 2024 18:03:24 +0530
Subject: [PATCH] Merging PR#327 into main branch

Refer https://github.com/ROCm/vllm/pull/327 for more info
---
 Dockerfile.rocm | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/Dockerfile.rocm b/Dockerfile.rocm
index 1f44958f5d1f1..fc07eb6297352 100644
--- a/Dockerfile.rocm
+++ b/Dockerfile.rocm
@@ -7,8 +7,8 @@ ARG BASE_IMAGE=rocm/vllm-dev:base_ubuntu22.04_py3.12_ROCm6.3_hipblaslt0.12_torch
 
 FROM ${BASE_IMAGE} AS base
 
-ARG ARG_PYTORCH_ROCM_ARCH
-ENV PYTORCH_ROCM_ARCH=${ARG_PYTORCH_ROCM_ARCH:-${PYTORCH_ROCM_ARCH}}
+ARG ARG_PYTORCH_ROCM_ARCH="gfx90a;gfx942;gfx1030;gfx1100;gfx1101"
+ENV PYTORCH_ROCM_ARCH=${ARG_PYTORCH_ROCM_ARCH}
 
 # Install some basic utilities
 RUN apt-get update -q -y && apt-get install -q -y \