llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[Bazel] Update to Ubuntu-22.04 and clang-16 for Bazel build docker (#1523) 

* Update Ubuntu and clang in the docker container
* Specifically build just `@torch-mlir//:torch-mlir-opt`


Triggered GHA run:
https://github.com/sjain-stanford/torch-mlir/actions/runs/3317006870/jobs/5479411204
pull/1575/head snapshot-20221111.654
Sambhav Jain 2022-11-10 13:11:06 -08:00 committed by GitHub
parent 6c31b06922
commit dcff5a7150
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 28 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
utils/bazel/.bazelrc
View File

@ -2,8 +2,8 @@
# See https://llvm.org/LICENSE.txt for license information. # See https://llvm.org/LICENSE.txt for license information.
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception # SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
build --action_env=CC=clang build --action_env=CC=clang-16
build --action_env=CXX=clang++ build --action_env=CXX=clang++-16
build --cxxopt=-std=c++17 build --cxxopt=-std=c++17
build --host_cxxopt=-std=c++17 build --host_cxxopt=-std=c++17
build --cxxopt=-D_GLIBCXX_USE_CXX11_ABI=0 build --cxxopt=-D_GLIBCXX_USE_CXX11_ABI=0

45
utils/bazel/docker/Dockerfile
View File

@ -1,35 +1,40 @@
ARG BASE_IMG=ubuntu:18.04 ARG BASE_IMG=ubuntu:22.04
FROM ${BASE_IMG} as dev-base FROM ${BASE_IMG} as dev-base
ARG ARCH="x86_64"
ARG BAZEL_VERSION=5.2.0
# Install basic packages # Install basic packages
RUN apt-get update && \ RUN apt-get update && \
apt-get install -y \ apt-get install -y \
clang-10 \ wget \
curl \ curl \
git \ git \
python3-pip \ python3-pip \
python3.8 \ python3.10 \
python3.8-dev \ python3.10-dev \
wget \
unzip unzip
RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.8 10 # Install clang
RUN update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.8 10 ARG REPO_NAME="deb http://apt.llvm.org/jammy/ llvm-toolchain-jammy main"
RUN echo $REPO_NAME >> /etc/apt/sources.list.d/llvm.list && \
RUN update-alternatives --install /usr/bin/clang clang /usr/bin/clang-10 10 wget -O - https://apt.llvm.org/llvm-snapshot.gpg.key | apt-key add - && \
RUN update-alternatives --install /usr/bin/clang++ clang++ /usr/bin/clang++-10 10 apt-get update && \
apt-get install -y \
clang
# Install bazel # Install bazel
ARG ARCH="x86_64"
ARG BAZEL_VERSION=5.2.0
RUN wget -q https://github.com/bazelbuild/bazel/releases/download/${BAZEL_VERSION}/bazel-${BAZEL_VERSION}-linux-${ARCH} -O /usr/bin/bazel \ RUN wget -q https://github.com/bazelbuild/bazel/releases/download/${BAZEL_VERSION}/bazel-${BAZEL_VERSION}-linux-${ARCH} -O /usr/bin/bazel \
&& chmod a+x /usr/bin/bazel && chmod a+x /usr/bin/bazel
# Install torch-mlir requirements
COPY requirements.txt /opt/app/requirements.txt COPY requirements.txt /opt/app/requirements.txt
COPY pytorch-requirements.txt /opt/app/pytorch-requirements.txt COPY pytorch-requirements.txt /opt/app/pytorch-requirements.txt
WORKDIR /opt/app WORKDIR /opt/app
RUN python -m pip install --upgrade pip RUN python3 -m pip install --upgrade pip
RUN python -m pip install --upgrade --ignore-installed -r requirements.txt RUN python3 -m pip install --upgrade --ignore-installed -r requirements.txt
# Clean up
RUN apt-get clean \
&& rm -rf /var/lib/apt/lists/*
WORKDIR /opt/src/torch-mlir WORKDIR /opt/src/torch-mlir

2
utils/bazel/docker/run_bazel_build.sh
View File

@ -1,3 +1,3 @@
#!/usr/bin/env bash #!/usr/bin/env bash
cd "$(pwd)/utils/bazel" && bazel build @torch-mlir//... cd "$(pwd)/utils/bazel" && bazel build @torch-mlir//:torch-mlir-opt