Update docker image

cornell-zhang · Feb 8, 2025 · 0db7f5f · 0db7f5f
1 parent ddebaba
commit 0db7f5f
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 7 deletions.
diff --git a/.github/workflows/config.yml b/.github/workflows/config.yml
@@ -15,7 +15,7 @@ jobs:
     # This job runs on Linux
     runs-on: ubuntu-latest
     container:
-      image: chhzh123/allo:llvm-19.x-py3.12
+      image: chhzh123/allo:latest
 
     steps:
     # https://github.com/actions/checkout/issues/363#issuecomment-1915075699

diff --git a/.github/workflows/sphinx_build.yml b/.github/workflows/sphinx_build.yml
@@ -23,7 +23,7 @@ jobs:
     # This job runs on Linux
     runs-on: ubuntu-latest
     container:
-      image: chhzh123/allo:llvm-19.x-py3.12
+      image: chhzh123/allo:latest
 
     steps:
     - name: Install rsync 📚

diff --git a/docker/demo.dockerfile b/docker/demo.dockerfile
@@ -49,7 +49,7 @@ RUN cd /root/llvm-project && \
     python3 -m pip install --upgrade pip && \
     python3 -m pip install numpy PyYAML dataclasses pybind11>=2.9.0  && \
     mkdir build && cd build && \
-    cmake -G Ninja ../llvm -DLLVM_ENABLE_PROJECTS=mlir \
+    cmake -G Ninja ../llvm -DLLVM_ENABLE_PROJECTS="clang;mlir;openmp" \
         -DLLVM_BUILD_EXAMPLES=ON -DLLVM_TARGETS_TO_BUILD="X86" \
         -DCMAKE_BUILD_TYPE=Release -DLLVM_ENABLE_ASSERTIONS=ON \
         -DLLVM_INSTALL_UTILS=ON -DMLIR_ENABLE_BINDINGS_PYTHON=ON \

diff --git a/docker/push.sh b/docker/push.sh
@@ -18,7 +18,7 @@ PASSWORD="$2"
 shift 1
 
 LOCAL_IMAGE_NAME=allo:latest
-REMOTE_IMAGE_NAME_VER=${DOCKER_HUB_ACCOUNT}/allo:llvm-19.x-py3.12
+REMOTE_IMAGE_NAME_VER=${DOCKER_HUB_ACCOUNT}/allo:latest
 
 echo "Login docker hub"
 docker login -u ${DOCKER_HUB_ACCOUNT} -p ${PASSWORD}

diff --git a/docs/source/setup/index.rst b/docs/source/setup/index.rst
@@ -34,8 +34,8 @@ To simplify the installation process, we provide a docker image that has already
 
 .. code-block:: console
 
-  $ docker pull chhzh123/allo:llvm-19.x-py3.12
-  $ docker run --rm -it chhzh123/allo:llvm-19.x-py3.12
+  $ docker pull chhzh123/allo:latest
+  $ docker run --rm -it chhzh123/allo:latest
   (docker) $ git clone https://github.com/cornell-zhang/allo.git && cd allo
   (docker) $ python3 -m pip install -v -e .
 
@@ -54,7 +54,7 @@ Please follow the instructions below to build the LLVM-19 project from source. Y
     # Python 3.12 is required
     mkdir -p build && cd build
     cmake -G Ninja ../llvm \
-        -DLLVM_ENABLE_PROJECTS=mlir \
+        -DLLVM_ENABLE_PROJECTS="clang;mlir;openmp" \
         -DLLVM_BUILD_EXAMPLES=ON \
         -DLLVM_TARGETS_TO_BUILD="host" \
         -DCMAKE_BUILD_TYPE=Release \