diff --git a/.buildkite/scripts/annotate-rocm-release.sh b/.buildkite/scripts/annotate-rocm-release.sh index 0a817890c..8a5b34440 100755 --- a/.buildkite/scripts/annotate-rocm-release.sh +++ b/.buildkite/scripts/annotate-rocm-release.sh @@ -68,7 +68,7 @@ aws s3 cp s3://${S3_BUCKET}/rocm/${BUILDKITE_COMMIT}/${ROCM_VERSION_PATH}/triton aws s3 cp s3://${S3_BUCKET}/rocm/${BUILDKITE_COMMIT}/${ROCM_VERSION_PATH}/torchvision-*.whl . aws s3 cp s3://${S3_BUCKET}/rocm/${BUILDKITE_COMMIT}/${ROCM_VERSION_PATH}/torchaudio-*.whl . aws s3 cp s3://${S3_BUCKET}/rocm/${BUILDKITE_COMMIT}/${ROCM_VERSION_PATH}/amdsmi-*.whl . -aws s3 cp s3://${S3_BUCKET}/rocm/${BUILDKITE_COMMIT}/${ROCM_VERSION_PATH}/aiter-*.whl . +aws s3 cp s3://${S3_BUCKET}/rocm/${BUILDKITE_COMMIT}/${ROCM_VERSION_PATH}/amd_aiter-*.whl . aws s3 cp s3://${S3_BUCKET}/rocm/${BUILDKITE_COMMIT}/${ROCM_VERSION_PATH}/flash-attn-*.whl . \`\`\` @@ -80,7 +80,7 @@ aws s3 cp s3://${S3_BUCKET}/rocm/${BUILDKITE_COMMIT}/${ROCM_VERSION_PATH}/flash- - **torchvision**: TorchVision for ROCm PyTorch - **torchaudio**: Torchaudio for ROCm PyTorch - **amdsmi**: AMD SMI Python bindings -- **aiter**: Aiter for ROCm +- **amd_aiter**: Aiter for ROCm - **flash-attn**: Flash Attention for ROCm ### :warning: Notes diff --git a/tools/vllm-rocm/pin_rocm_dependencies.py b/tools/vllm-rocm/pin_rocm_dependencies.py index b9387069d..7d90d6669 100644 --- a/tools/vllm-rocm/pin_rocm_dependencies.py +++ b/tools/vllm-rocm/pin_rocm_dependencies.py @@ -64,7 +64,7 @@ def get_custom_wheel_versions(install_dir: str) -> dict[str, str]: ("torchaudio-", "torchaudio"), # Match torchaudio- ("amdsmi-", "amdsmi"), # Match amdsmi- ("flash_attn-", "flash-attn"), # Match flash_attn- - ("aiter-", "aiter"), # Match aiter- + ("amd_aiter-", "amd-aiter"), # Match amd_aiter- ] for wheel_file in install_path.glob("*.whl"):