From 57ee730cc6fd6a22c7469617af76bc6dc161d98d Mon Sep 17 00:00:00 2001
From: Debasis Mandal <debasis.mandal@amd.com>
Date: Wed, 24 Jun 2026 14:58:44 +0000
Subject: [PATCH] fix(devcontainer): align AITER ROCm index with PyTorch ROCm
 version

Default AITER_ROCM_VERSION to ROCM_VERSION so the devcontainer installs
both packages from the same ROCm release, avoiding binary mismatches.

Co-authored-by: Cursor <cursoragent@cursor.com>
---
 .devcontainer/rocm/Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.devcontainer/rocm/Dockerfile b/.devcontainer/rocm/Dockerfile
index 0f0fbe84cd..b1c4f6381e 100644
--- a/.devcontainer/rocm/Dockerfile
+++ b/.devcontainer/rocm/Dockerfile
@@ -8,7 +8,7 @@ ARG ROCM_VERSION
 ARG PY_VERSION=3.12
 ARG TORCH_VERSION=2.9.1
 ARG AITER_VERSION=0.1.10
-ARG AITER_ROCM_VERSION=7.1.1
+ARG AITER_ROCM_VERSION=${ROCM_VERSION}
 
 # Update package lists and install system dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \