ppc64le yml dockefile

Author: sandeepgupta12

.github/scripts/ppc64le-build.sh

@@ -0,0 +1,47 @@
+# Dockerfile.ppc64le
+FROM registry.access.redhat.com/ubi9/ubi:9.3
+
+LABEL maintainer="Md. Shafi Hussain <[email protected]>"
+
+# Install necessary dependencies
+RUN dnf install -y \
+    https://dl.fedoraproject.org/pub/epel/epel-release-latest-9.noarch.rpm && \
+    dnf install -y git cmake ninja-build gcc-c++ rust cargo \
+    python39 python39-devel python39-wheel python39-pip python39-setuptools && \
+    dnf clean all
+
+# Link Python and pip to expected paths if not available
+RUN if ! command -v python; then ln -s $(command -v python3.9) /usr/bin/python; fi && \
+    if ! command -v pip; then ln -s $(command -v pip3.9) /usr/bin/pip; fi
+
+# Environment variables
+ENV PACKAGE_NAME=pytorch \
+    PACKAGE_VERSION=v2.4.0 \
+    PACKAGE_URL=https://github.com/pytorch/pytorch.git \
+    _GLIBCXX_USE_CXX11_ABI=1
+
+# Clone PyTorch repository and apply patch if necessary
+RUN git clone $PACKAGE_URL /workspace/$PACKAGE_NAME && \
+    cd /workspace/$PACKAGE_NAME && \
+    git checkout tags/$PACKAGE_VERSION && \
+    PPC64LE_PATCH="69cbf05" && \
+    if ! git log --pretty=format:"%H" | grep -q "$PPC64LE_PATCH"; then \
+        git config user.email "[email protected]" && \
+        git config user.name "Md. Shafi Hussain" && \
+        git cherry-pick "$PPC64LE_PATCH"; \
+    else \
+        echo "POWER patch not needed."; \
+    fi
+
+# Set up submodules and install dependencies
+RUN cd /workspace/$PACKAGE_NAME && \
+    git submodule sync && \
+    git submodule update --init --recursive && \
+    pip install -r requirements.txt
+
+# Copy the build script to the container
+COPY .github/scripts/ppc64le-build.sh /workspace/.github/scripts/ppc64le-build.sh
+RUN chmod +x /workspace/.github/scripts/ppc64le-build.sh
+
+# Default command
+CMD ["/workspace/.github/scripts/ppc64le-build.sh"]

.github/workflows/Dockerfile.ppc64le

@@ -17,61 +17,14 @@ jobs:
       - name: Checkout repository
         uses: actions/checkout@v3
 
-      - name: Set up Python
-        uses: actions/setup-python@v4
-        with:
-          python-version: 3.9  # Adjust if a different Python version is needed
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v2
 
-      - name: Install dependencies
+      - name: Build Docker image for ppc64le
         run: |
-          # Install EPEL release
-          sudo dnf install -y https://dl.fedoraproject.org/pub/epel/epel-release-latest-9.noarch.rpm
-          
-          # Install development tools and libraries
-          sudo dnf install -y git cmake ninja-build g++ rust cargo \
-              python3.9-devel python3.9-wheel python3.9-pip python3.9-setuptools
-              
-          # Ensure Python and pip are accessible by symlinking
-          if ! command -v python; then
-              sudo ln -s $(command -v python3.9) /usr/bin/python
-          fi
-          if ! command -v pip; then
-              sudo ln -s $(command -v pip3.9) /usr/bin/pip
-          fi
+          docker build -f .github/workflows/Dockerfile.ppc64le -t pytorch-ppc64le:ubi9.3 .
 
-      - name: Clone and Checkout
+      - name: Run Docker container and execute build script
         run: |
-          # Clone the repository and check out the specified version
-          git clone https://github.com/pytorch/pytorch.git
-          cd pytorch
-          git checkout tags/v2.4.0
-
-      - name: Apply POWER Patch
-        run: |
-          cd pytorch
-          PPC64LE_PATCH="69cbf05"
-          if ! git log --pretty=format:"%H" | grep -q "$PPC64LE_PATCH"; then
-              echo "Applying POWER patch."
-              git config user.email "[email protected]"
-              git config user.name "Md. Shafi Hussain"
-              git cherry-pick "$PPC64LE_PATCH"
-          else
-              echo "POWER patch not needed."
-          fi
-
-      - name: Update submodules
-        run: |
-          cd pytorch
-          git submodule sync
-          git submodule update --init --recursive
-
-      - name: Install Python dependencies
-        run: |
-          cd pytorch
-          pip install -r requirements.txt
-
-      - name: Build PyTorch wheel
-        run: |
-          cd pytorch
-          MAX_JOBS=$(nproc) python setup.py bdist_wheel
-          pip install dist/*.whl
+          docker run --rm pytorch-ppc64le:ubi9.3 /workspace/.github/scripts/ppc64le-build.sh
+