| #!/bin/bash |
| # Copyright 2023 The IREE Authors |
| # |
| # Licensed under the Apache License v2.0 with LLVM Exceptions. |
| # See https://llvm.org/LICENSE.txt for license information. |
| # SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| |
| set -euo pipefail |
| |
| ROCM_VERSION=$1 |
| AMDGPU_VERSION=$2 |
| |
| ARCH="$(uname -m)" |
| if [[ "${ARCH}" == "x86_64" ]]; then |
| apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends ca-certificates curl libnuma-dev gnupg \ |
| && curl -sL https://repo.radeon.com/rocm/rocm.gpg.key | apt-key add - \ |
| && printf "deb [arch=amd64] https://repo.radeon.com/rocm/apt/${ROCM_VERSION}/ ubuntu main" | tee /etc/apt/sources.list.d/rocm.list \ |
| && printf "deb [arch=amd64] https://repo.radeon.com/amdgpu/${AMDGPU_VERSION}/ubuntu focal main" | tee /etc/apt/sources.list.d/amdgpu.list \ |
| && apt-get update \ |
| && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \ |
| libelf1 \ |
| kmod \ |
| file \ |
| rocm-dev \ |
| build-essential |
| else |
| echo "Installing ROCM for ${ARCH} is not supported yet." |
| fi |
