Redesign

.devcontainer/Dockerfile

@@ -0,0 +1,148 @@
+ARG BASE=mcr.microsoft.com/devcontainers/base:ubuntu-20.04
+
+ARG LLVM_VERSION=15.0.7
+ARG HETEROCL_VERSION=69181b080c8cf7d9fab0a6b2ae3511923274b32e
+ARG HCL_DIALECT_VERSION=f2f5e43a93173f9a86671c8f5004d5fb21cb329b
+
+FROM $BASE AS build
+
+ARG LLVM_VERSION
+ARG HETEROCL_VERSION
+ARG HCL_DIALECT_VERSION
+ARG DEBIAN_FRONTEND=noninteractive
+
+RUN apt-get update && \
+    apt-get install -y \
+        apt-transport-https \
+        ca-certificates \
+        g++ \
+        gcc \
+        git \
+        gnupg \
+        make \
+        python3-dev \
+        python3-pip \
+        python3-venv \
+        python3-virtualenv \
+        software-properties-common \
+        wget \
+        xz-utils
+
+RUN wget -O - https://apt.kitware.com/keys/kitware-archive-latest.asc 2>/dev/null | apt-key add - && \
+    apt-add-repository 'deb https://apt.kitware.com/ubuntu/ focal main' && \
+    apt-get update -y && \
+    apt-get install -y cmake
+
+RUN pip install -U \
+        pip \
+        build \
+        patchelf \
+        auditwheel
+
+# Download LLVM
+ARG LLVM_DIR=/opt/llvm-project
+ADD "https://github.com/llvm/llvm-project/releases/download/llvmorg-$LLVM_VERSION/llvm-project-$LLVM_VERSION.src.tar.xz" /tmp/
+RUN cd /tmp && \
+    tar -xf llvm-project-$LLVM_VERSION.src.tar.xz && \
+    mv llvm-project-$LLVM_VERSION.src $LLVM_DIR && \
+    rm llvm-project-$LLVM_VERSION.src.tar.xz
+
+# Build LLVM & MLIR
+WORKDIR $LLVM_DIR/build
+RUN pip install -r ../mlir/python/requirements.txt
+RUN cmake ../llvm \
+        -DCMAKE_BUILD_TYPE=Release \
+        -DLLVM_ENABLE_PROJECTS=mlir \
+        -DLLVM_TARGETS_TO_BUILD=Native \
+        -DLLVM_ENABLE_ASSERTIONS=ON \
+        -DLLVM_INSTALL_UTILS=ON \
+        -DMLIR_ENABLE_BINDINGS_PYTHON=ON \
+        && \
+    make -j`nproc`
+
+# Download HeteroCL
+ARG HETEROCL_DIR=/opt/heterocl
+ARG HCL_DIALECT_DIR=$HETEROCL_DIR/hcl-dialect
+RUN git clone https://github.com/cornell-zhang/heterocl.git $HETEROCL_DIR && \
+    cd $HETEROCL_DIR && \
+    git checkout $HETEROCL_VERSION
+RUN git clone https://github.com/cornell-zhang/hcl-dialect.git $HCL_DIALECT_DIR && \
+    cd $HCL_DIALECT_DIR && \
+    git checkout $HCL_DIALECT_VERSION
+
+# Build hcl-dialect
+WORKDIR $HCL_DIALECT_DIR/build
+RUN cmake .. \
+        -DMLIR_DIR=$LLVM_DIR/build/lib/cmake/mlir \
+        -DLLVM_EXTERNAL_LIT=$LLVM_DIR/build/bin/llvm-lit \
+        -DPYTHON_BINDING=ON \
+        -DOPENSCOP=OFF \
+        && \
+    make -j`nproc`
+
+RUN pip install hatch && \
+    mkdir -p /opt/wheelhouse
+
+# Package hcl-dialect
+WORKDIR $HCL_DIALECT_DIR/build/tools/hcl/python_packages/hcl_core
+RUN rm setup.py && \
+    cat <<EOF > pyproject.toml
+[project]
+name = "hcl_mlir"
+requires-python = ">= 3.6"
+version = "0.1"
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[tool.hatch.build]
+ignore-vcs = true
+
+[tool.hatch.build.targets.wheel]
+pure_python = false
+packages = ["hcl_mlir"]
+EOF
+RUN hatch build && \
+    mv dist/*.whl /opt/wheelhouse
+
+# Package HeteroCL
+WORKDIR $HETEROCL_DIR
+RUN rm setup.py && \
+    cat <<EOF > pyproject.toml
+[project]
+name = "heterocl"
+requires-python = ">= 3.6"
+version = "0.5"
+dynamic = ["dependencies"]
+
+[build-system]
+requires = ["hatchling", "hatch-requirements-txt"]
+build-backend = "hatchling.build"
+
+[tool.hatch.build]
+ignore-vcs = true
+
+[tool.hatch.build.targets.wheel]
+pure_python = false
+packages = ["heterocl"]
+
+[tool.hatch.metadata.hooks.requirements_txt]
+files = ["requirements.txt"]
+EOF
+RUN hatch build && \
+    mv dist/*.whl /opt/wheelhouse
+
+
+FROM $BASE
+
+ARG DEBIAN_FRONTEND=noninteractive
+
+COPY --from=build /opt/wheelhouse /opt/wheelhouse
+
+RUN apt-get update && \
+    apt-get install -y \
+        python3-pip \
+        && \
+    pip install attrs /opt/wheelhouse/*.whl && \
+    rm -r /var/lib/apt/lists/* /opt/wheelhouse

.devcontainer/devcontainer.json

@@ -0,0 +1,24 @@
+// For format details, see https://aka.ms/devcontainer.json. For config options, see the
+// README at: https://github.com/devcontainers/templates/tree/main/src/miniconda
+{
+	"name": "Ubuntu",
+	"build": {
+		"context": "..",
+		"dockerfile": "Dockerfile"
+	}
+
+	// Features to add to the dev container. More info: https://containers.dev/features.
+	// "features": {},
+
+	// Use 'forwardPorts' to make a list of ports inside the container available locally.
+	// "forwardPorts": [],
+
+	// Use 'postCreateCommand' to run commands after the container is created.
+	// "postCreateCommand": "python --version",
+
+	// Configure tool-specific properties.
+	// "customizations": {},
+
+	// Uncomment to connect as root instead. More info: https://aka.ms/dev-containers-non-root.
+	// "remoteUser": "root"
+}