Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
59 commits
Select commit Hold shift + click to select a range
4a738c0
removed pytorch permission to run the ci locally
sandeepgupta12 Mar 4, 2025
e1baf34
nightly build changes
sandeepgupta12 Mar 10, 2025
1bfd1fe
nightly build changes
sandeepgupta12 Mar 10, 2025
2c0808d
nightly build changes
sandeepgupta12 Mar 11, 2025
a798afa
nightly build changes
sandeepgupta12 Mar 11, 2025
4179dd0
nightly build changes
sandeepgupta12 Mar 11, 2025
6b15b97
nightly build changes
sandeepgupta12 Mar 11, 2025
9dd0f85
nightly build changes
sandeepgupta12 Mar 11, 2025
91462d3
nightly build changes
sandeepgupta12 Mar 11, 2025
7c864b8
nightly build changes
sandeepgupta12 Mar 11, 2025
ab6048c
nightly build changes
sandeepgupta12 Mar 11, 2025
3b298ea
nightly build changes
sandeepgupta12 Mar 11, 2025
2f21bc6
nightly build changes
sandeepgupta12 Mar 11, 2025
18d339c
nightly build changes
sandeepgupta12 Mar 11, 2025
30cb77e
nightly build changes
sandeepgupta12 Mar 11, 2025
1e75325
nightly build changes
sandeepgupta12 Mar 11, 2025
2596686
nightly build changes
sandeepgupta12 Mar 12, 2025
6e4db3f
nightly build changes
sandeepgupta12 Mar 12, 2025
0d53da1
nightly build changes
sandeepgupta12 Mar 12, 2025
c960a1a
nightly build changes
sandeepgupta12 Mar 12, 2025
8827de2
nightly build changes
sandeepgupta12 Mar 12, 2025
a67486c
nightly build changes
sandeepgupta12 Mar 12, 2025
96e9fc1
nightly build changes
sandeepgupta12 Mar 12, 2025
a33f481
nightly build changes
sandeepgupta12 Mar 13, 2025
f429418
nightly build changes
sandeepgupta12 Mar 13, 2025
d8ff46f
nightly build changes p311
sandeepgupta12 Mar 13, 2025
46cb4f8
nightly build changes p312 and 313
sandeepgupta12 Mar 13, 2025
cd6f2dc
nightly build changes p312 and 313
sandeepgupta12 Mar 13, 2025
61ec5eb
nightly build changes p312 and 313
sandeepgupta12 Mar 13, 2025
f8df963
nightly build changes p312 and 313
sandeepgupta12 Mar 13, 2025
a3b2543
nightly build changes p312 and 313
sandeepgupta12 Mar 13, 2025
de290a4
nightly build changes p312 and 313
sandeepgupta12 Mar 13, 2025
af8d3af
nightly build changes 313
sandeepgupta12 Mar 13, 2025
8c45f06
nightly build changes except 313
sandeepgupta12 Mar 13, 2025
f63765f
nightly build changes except 313
sandeepgupta12 Mar 18, 2025
b1b9abe
nightly build changes except 313
sandeepgupta12 Mar 18, 2025
853b909
nightly build changes except 313
sandeepgupta12 Mar 18, 2025
a4f694b
nightly build changes except 313
sandeepgupta12 Mar 18, 2025
f0f4a8c
nightly build changes except 313
sandeepgupta12 Mar 18, 2025
38702e5
nightly build changes except 313
sandeepgupta12 Mar 18, 2025
e0c5cc5
nightly build changes except 313
sandeepgupta12 Mar 18, 2025
634f946
nightly build changes except 313
sandeepgupta12 Mar 18, 2025
9136a39
nightly build changes except 313
sandeepgupta12 Mar 18, 2025
86a3449
nightly build changes except 313
sandeepgupta12 Mar 18, 2025
d133a43
nightly build changes except 313
sandeepgupta12 Mar 18, 2025
49c382e
nightly build changes except 313
sandeepgupta12 Mar 18, 2025
8afd505
nightly build changes except 313
sandeepgupta12 Mar 18, 2025
bccf82e
nightly build changes except 313
sandeepgupta12 Mar 18, 2025
b920fe1
nightly build changes except 313
sandeepgupta12 Mar 18, 2025
ce7a878
nightly build changes except 313
sandeepgupta12 Mar 18, 2025
f9b7831
nightly build changes except 313
sandeepgupta12 Mar 18, 2025
7c2d9c3
nightly build changes except 313
sandeepgupta12 Mar 18, 2025
71a77f6
nightly build changes except 313
sandeepgupta12 Mar 18, 2025
ce6202e
nightly build changes except 313
sandeepgupta12 Mar 19, 2025
c5f3ba8
nightly build changes except 313
sandeepgupta12 Mar 19, 2025
08c0b96
nightly build changes except 313
sandeepgupta12 Mar 19, 2025
e4a141b
nightly build changes except 313
sandeepgupta12 Mar 19, 2025
5255d32
nightly build changes except 313
sandeepgupta12 Mar 20, 2025
ab68338
Merge branch 'temp-nightly-1' into temp-ppc64le-wheel-branch-v6
sandeepgupta12 Mar 20, 2025
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
33 changes: 31 additions & 2 deletions .ci/docker/manywheel/Dockerfile_ppc64le
Original file line number Diff line number Diff line change
Expand Up @@ -75,9 +75,38 @@ ENV LD_LIBRARY_PATH=/opt/rh/gcc-toolset-${DEVTOOLSET_VERSION}/root/usr/lib64:/op
# Configure git to avoid safe directory issues
RUN git config --global --add safe.directory "*"

# installed python doesn't have development parts. Rebuild it from scratch
RUN /bin/rm -rf /opt/_internal /opt/python /usr/local/*/*

# EPEL for cmake
FROM base as patchelf
# Install patchelf
ADD ./common/install_patchelf.sh install_patchelf.sh
RUN bash ./install_patchelf.sh && rm install_patchelf.sh
RUN cp $(which patchelf) /patchelf

FROM patchelf as python
# build python
COPY manywheel/build_scripts /build_scripts
ADD ./common/install_cpython.sh /build_scripts/install_cpython.sh
ENV SSL_CERT_FILE=
RUN bash build_scripts/build.sh && rm -r build_scripts
#RUN bash build_scripts/build.sh || (echo "Checksum verification failed!" && exit 1)

FROM base as final
COPY --from=python /opt/python /opt/python
COPY --from=python /opt/_internal /opt/_internal
COPY --from=python /opt/python/cp39-cp39/bin/auditwheel /usr/local/bin/auditwheel
COPY --from=patchelf /usr/local/bin/patchelf /usr/local/bin/patchelf

RUN alternatives --set python /usr/bin/python3.12
RUN alternatives --set python3 /usr/bin/python3.12

RUN pip-3.12 install typing_extensions

# Install required Python packages
RUN pip install --upgrade pip
RUN pip install typing_extensions pyyaml setuptools
#RUN pip install --upgrade pip
#RUN pip install typing_extensions pyyaml setuptools

# Install test dependencies
RUN dnf install -y \
Expand Down
2 changes: 1 addition & 1 deletion .ci/docker/manywheel/build.sh
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,7 @@ case ${GPU_ARCH_TYPE} in
MANY_LINUX_VERSION="s390x"
;;
cpu-ppc64le)
TARGET=base
TARGET=final
DOCKER_TAG=ppc64le
GPU_IMAGE=redhat/ubi9
DOCKER_GPU_BUILD_ARG=""
Expand Down
29 changes: 27 additions & 2 deletions .ci/docker/manywheel/build_scripts/build.sh
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@ DEVTOOLS_HASH=a8ebeb4bed624700f727179e6ef771dafe47651131a00a78b342251415646acc
PATCHELF_HASH=d9afdff4baeacfbc64861454f368b7f2c15c44d245293f7587bbf726bfe722fb
CURL_ROOT=curl-7.73.0
CURL_HASH=cf34fe0b07b800f1c01a499a6e8b2af548f6d0e044dca4a29d88a4bee146d131
#CURL_HASH=ed444155f1fd7d72c44424f9333893e64cbd6c1cbe4489619bce99c4dda58c14
AUTOCONF_ROOT=autoconf-2.69
AUTOCONF_HASH=954bd69b391edc12d6a4a51a2dd1476543da5c6bbf05a95b59dc0dd6fd4c2969

Expand All @@ -38,8 +39,32 @@ yum -y install bzip2 make git patch unzip bison yasm diffutils \
automake which file \
${PYTHON_COMPILE_DEPS}

# Install newest autoconf
build_autoconf $AUTOCONF_ROOT $AUTOCONF_HASH
# Download and extract Autoconf
curl -sLO http://ftp.gnu.org/gnu/autoconf/$AUTOCONF_ROOT.tar.gz
echo "$AUTOCONF_HASH $AUTOCONF_ROOT.tar.gz" | sha256sum -c -
tar -xzf $AUTOCONF_ROOT.tar.gz
cd $AUTOCONF_ROOT

# Update config.guess and config.sub
curl -o build-aux/config.guess http://git.savannah.gnu.org/cgit/config.git/plain/config.guess
curl -o build-aux/config.sub http://git.savannah.gnu.org/cgit/config.git/plain/config.sub
chmod +x build-aux/config.guess build-aux/config.sub

# Configure with the correct host
./configure --host=powerpc64le-pc-linux-gnu

# Build and install
make -j$(nproc)
make install

# Clean up
cd ..
rm -rf $AUTOCONF_ROOT $AUTOCONF_ROOT.tar.gz


# Install newest autoconf (previous method)
#build_autoconf $AUTOCONF_ROOT $AUTOCONF_HASH

autoconf --version

# Compile the latest Python releases.
Expand Down
10 changes: 8 additions & 2 deletions .ci/docker/manywheel/build_scripts/build_utils.sh
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,12 @@ function check_sha256sum {
check_var ${fname}
local sha256=$2
check_var ${sha256}

# Compute and print actual checksum
echo "✅ Expected SHA256: ${sha256}"
echo "🔍 Calculating actual SHA256..."
actual_sha256=$(sha256sum ${fname} | awk '{print $1}')
echo "🔴 Actual SHA256: $actual_sha256"

echo "${sha256} ${fname}" > ${fname}.sha256
sha256sum -c ${fname}.sha256
rm -f ${fname}.sha256
Expand All @@ -41,7 +46,8 @@ function build_openssl {
local openssl_sha256=$2
check_var ${openssl_sha256}
check_var ${OPENSSL_DOWNLOAD_URL}
curl -sLO ${OPENSSL_DOWNLOAD_URL}/${openssl_fname}.tar.gz
curl -sLO ${OPENSSL_DOWNLOAD_URL}/${openssl_fname}.tar.gz

check_sha256sum ${openssl_fname}.tar.gz ${openssl_sha256}
tar -xzf ${openssl_fname}.tar.gz
(cd ${openssl_fname} && do_openssl_build)
Expand Down
2 changes: 1 addition & 1 deletion .ci/docker/manywheel/build_scripts/manylinux1-check.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ def is_manylinux1_compatible():
# Only Linux, and only x86-64 / i686
from distutils.util import get_platform

if get_platform() not in ["linux-x86_64", "linux-i686", "linux-s390x"]:
if get_platform() not in ["linux-x86_64", "linux-i686", "linux-s390x", "linux-pc64le"]:
return False

# Check for presence of _manylinux module
Expand Down
2 changes: 1 addition & 1 deletion .ci/manywheel/build.sh
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@ case "${GPU_ARCH_TYPE:-BLANK}" in
rocm)
bash "${SCRIPTPATH}/build_rocm.sh"
;;
cpu | cpu-cxx11-abi | cpu-s390x)
cpu | cpu-cxx11-abi | cpu-s390x | cpu-ppc64le)
bash "${SCRIPTPATH}/build_cpu.sh"
;;
xpu)
Expand Down
4 changes: 2 additions & 2 deletions .ci/manywheel/build_common.sh
Original file line number Diff line number Diff line change
Expand Up @@ -380,7 +380,7 @@ for pkg in /$WHEELHOUSE_DIR/torch_no_python*.whl /$WHEELHOUSE_DIR/torch*linux*.w
done

# create Manylinux 2_28 tag this needs to happen before regenerate the RECORD
if [[ $PLATFORM == "manylinux_2_28_x86_64" && $GPU_ARCH_TYPE != "cpu-s390x" && $GPU_ARCH_TYPE != "xpu" ]]; then
if [[ $PLATFORM == "manylinux_2_28_x86_64" && $GPU_ARCH_TYPE != "cpu-s390x" && $GPU_ARCH_TYPE != "cpu-ppc64le" && $GPU_ARCH_TYPE != "xpu" ]]; then
wheel_file=$(echo $(basename $pkg) | sed -e 's/-cp.*$/.dist-info\/WHEEL/g')
sed -i -e s#linux_x86_64#"${PLATFORM}"# $wheel_file;
fi
Expand Down Expand Up @@ -425,7 +425,7 @@ for pkg in /$WHEELHOUSE_DIR/torch_no_python*.whl /$WHEELHOUSE_DIR/torch*linux*.w
fi

# Rename wheel for Manylinux 2_28
if [[ $PLATFORM == "manylinux_2_28_x86_64" && $GPU_ARCH_TYPE != "cpu-s390x" && $GPU_ARCH_TYPE != "xpu" ]]; then
if [[ $PLATFORM == "manylinux_2_28_x86_64" && $GPU_ARCH_TYPE != "cpu-s390x" && $GPU_ARCH_TYPE != "cpu-ppc64le" && $GPU_ARCH_TYPE != "xpu" ]]; then
pkg_name=$(echo $(basename $pkg) | sed -e s#linux_x86_64#"${PLATFORM}"#)
zip -rq $pkg_name $PREIX*
rm -f $pkg
Expand Down
2 changes: 2 additions & 0 deletions .ci/manywheel/build_cpu.sh
Original file line number Diff line number Diff line change
Expand Up @@ -36,6 +36,8 @@ elif [[ "$OS_NAME" == *"AlmaLinux"* ]]; then
elif [[ "$OS_NAME" == *"Ubuntu"* ]]; then
if [[ "$(uname -m)" == "s390x" ]]; then
LIBGOMP_PATH="/usr/lib/s390x-linux-gnu/libgomp.so.1"
elif [[ "$(uname -m)" == "ppc64le" ]]; then
LIBGOMP_PATH="/usr/lib64/libgomp.so.1"
else
LIBGOMP_PATH="/usr/lib/x86_64-linux-gnu/libgomp.so.1"
fi
Expand Down
6 changes: 3 additions & 3 deletions .ci/pytorch/check_binary.sh
Original file line number Diff line number Diff line change
Expand Up @@ -263,7 +263,7 @@ fi
if [[ "$PACKAGE_TYPE" == 'libtorch' ]]; then
echo "Checking that MKL is available"
build_and_run_example_cpp check-torch-mkl
elif [[ "$(uname -m)" != "arm64" && "$(uname -m)" != "s390x" ]]; then
elif [[ "$(uname -m)" != "arm64" && "$(uname -m)" != "s390x" && "$(uname -m)" != "ppc64le" ]]; then
if [[ "$(uname)" != 'Darwin' || "$PACKAGE_TYPE" != *wheel ]]; then
if [[ "$(uname -m)" == "aarch64" ]]; then
echo "Checking that MKLDNN is available on aarch64"
Expand All @@ -287,7 +287,7 @@ if [[ "$PACKAGE_TYPE" == 'libtorch' ]]; then
echo "Checking that XNNPACK is available"
build_and_run_example_cpp check-torch-xnnpack
else
if [[ "$(uname)" != 'Darwin' || "$PACKAGE_TYPE" != *wheel ]] && [[ "$(uname -m)" != "s390x" ]]; then
if [[ "$(uname)" != 'Darwin' || "$PACKAGE_TYPE" != *wheel ]] && [[ "$(uname -m)" != "s390x" ]] && [[ "$(uname -m)" != "ppc64le" ]]; then
echo "Checking that XNNPACK is available"
pushd /tmp
python -c 'import torch.backends.xnnpack; exit(0 if torch.backends.xnnpack.enabled else 1)'
Expand All @@ -308,7 +308,7 @@ if [[ "$OSTYPE" == "msys" ]]; then
fi

# Test that CUDA builds are setup correctly
if [[ "$DESIRED_CUDA" != 'cpu' && "$DESIRED_CUDA" != 'xpu' && "$DESIRED_CUDA" != 'cpu-cxx11-abi' && "$DESIRED_CUDA" != *"rocm"* && "$(uname -m)" != "s390x" ]]; then
if [[ "$DESIRED_CUDA" != 'cpu' && "$DESIRED_CUDA" != 'xpu' && "$DESIRED_CUDA" != 'cpu-cxx11-abi' && "$DESIRED_CUDA" != *"rocm"* && "$(uname -m)" != "s390x" && "$(uname -m)" != "ppc64le" ]]; then
if [[ "$PACKAGE_TYPE" == 'libtorch' ]]; then
build_and_run_example_cpp check-torch-cuda
else
Expand Down
4 changes: 2 additions & 2 deletions .github/actions/test-pytorch-binary/action.yml
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,7 @@ runs:
)

echo "CONTAINER_NAME=${container_name}" >> "$GITHUB_ENV"
if [[ "${GPU_ARCH_TYPE}" != "rocm" && "${BUILD_ENVIRONMENT}" != "linux-aarch64-binary-manywheel" && "${BUILD_ENVIRONMENT}" != "linux-s390x-binary-manywheel" && "${GPU_ARCH_TYPE}" != "xpu" ]]; then
if [[ "${GPU_ARCH_TYPE}" != "rocm" && "${BUILD_ENVIRONMENT}" != "linux-aarch64-binary-manywheel" && "${BUILD_ENVIRONMENT}" != "linux-s390x-binary-manywheel" && "${BUILD_ENVIRONMENT}" != "linux-ppc64le-binary-manywheel" && "${GPU_ARCH_TYPE}" != "xpu" ]]; then
# Propagate download.pytorch.org IP to container. This is only needed on Linux non aarch64 runner
grep download.pytorch.org /etc/hosts | docker exec -i "${container_name}" bash -c "/bin/cat >> /etc/hosts"
fi
Expand All @@ -46,7 +46,7 @@ runs:
docker exec -t "${container_name}" bash -c "source ${BINARY_ENV_FILE} && bash -x /run.sh"

- name: Cleanup docker
if: always() && (env.BUILD_ENVIRONMENT == 'linux-s390x-binary-manywheel' || env.GPU_ARCH_TYPE == 'xpu')
if: always() && (env.BUILD_ENVIRONMENT == 'linux-s390x-binary-manywheel' || env.BUILD_ENVIRONMENT != 'linux-ppc64le-binary-manywheel' || env.GPU_ARCH_TYPE == 'xpu')
shell: bash
run: |
# on s390x or xpu stop the container for clean worker stop
Expand Down
24 changes: 19 additions & 5 deletions .github/scripts/generate_binary_build_matrix.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,6 +40,9 @@

CPU_S390X_ARCH = ["cpu-s390x"]


CPU_PPC64LE_ARCH = ["cpu-ppc64le"]

CUDA_AARCH64_ARCHES = ["12.8-aarch64"]


Expand Down Expand Up @@ -152,6 +155,8 @@ def arch_type(arch_version: str) -> str:
return "cpu-aarch64"
elif arch_version in CPU_S390X_ARCH:
return "cpu-s390x"
elif arch_version in CPU_PPC64LE_ARCH:
return "cpu-ppc64le"
elif arch_version in CUDA_AARCH64_ARCHES:
return "cuda-aarch64"
else: # arch_version should always be "cpu" in this case
Expand Down Expand Up @@ -179,6 +184,7 @@ def arch_type(arch_version: str) -> str:
"cpu-cxx11-abi": f"pytorch/manylinuxcxx11-abi-builder:cpu-cxx11-abi-{DEFAULT_TAG}",
"cpu-aarch64": f"pytorch/manylinux2_28_aarch64-builder:cpu-aarch64-{DEFAULT_TAG}",
"cpu-s390x": f"pytorch/manylinuxs390x-builder:cpu-s390x-{DEFAULT_TAG}",
"cpu-ppc64le": f"pytorch/manylinuxppc64le-builder:cpu-ppc64le-{DEFAULT_TAG}",
}

CXX11_ABI = "cxx11-abi"
Expand Down Expand Up @@ -212,6 +218,7 @@ def translate_desired_cuda(gpu_arch_type: str, gpu_arch_version: str) -> str:
"cpu-aarch64": "cpu",
"cpu-cxx11-abi": "cpu-cxx11-abi",
"cpu-s390x": "cpu",
"cpu-ppc64le": "cpu",
"cuda": f"cu{gpu_arch_version.replace('.', '')}",
"cuda-aarch64": f"cu{gpu_arch_version.replace('-aarch64', '').replace('.', '')}",
"rocm": f"rocm{gpu_arch_version}",
Expand Down Expand Up @@ -290,8 +297,8 @@ def generate_wheels_matrix(
use_split_build: bool = False,
) -> list[dict[str, str]]:
package_type = "wheel"
if os == "linux" or os == "linux-aarch64" or os == "linux-s390x":
# NOTE: We only build manywheel packages for x86_64 and aarch64 and s390x linux
if os == "linux" or os == "linux-aarch64" or os == "linux-s390x" or os == "linux-ppc64le":
# NOTE: We only build manywheel packages for x86_64 and aarch64 and s390x and ppc64le linux
package_type = "manywheel"

if python_versions is None:
Expand All @@ -312,6 +319,11 @@ def generate_wheels_matrix(
# Only want the one arch as the CPU type is different and
# uses different build/test scripts
arches = ["cpu-s390x"]
elif os == "linux-ppc64le":
# Only want the one arch as the CPU type is different and
# uses different build/test scripts
arches = ["cpu-ppc64le"]


ret: list[dict[str, str]] = []
for python_version in python_versions:
Expand All @@ -323,13 +335,15 @@ def generate_wheels_matrix(
or arch_version == "cpu-cxx11-abi"
or arch_version == "cpu-aarch64"
or arch_version == "cpu-s390x"
or arch_version == "cpu-ppc64le"
or arch_version == "xpu"
else arch_version
)

# TODO: Enable python 3.13t on cpu-s390x
if gpu_arch_type == "cpu-s390x" and python_version == "3.13t":
continue

# TODO: Enable python 3.13t cpu-s390x or MacOS or Windows
if (gpu_arch_type == "cpu-s390x" or gpu_arch_type == "cpu-ppc64le") and python_version == "3.13t":
continue

if use_split_build and (
arch_version not in ["12.6", "12.8", "11.8", "cpu"] or os != "linux"
Expand Down
19 changes: 19 additions & 0 deletions .github/scripts/generate_ci_workflows.py
Original file line number Diff line number Diff line change
Expand Up @@ -101,6 +101,7 @@ class OperatingSystem:
MACOS_ARM64 = "macos-arm64"
LINUX_AARCH64 = "linux-aarch64"
LINUX_S390X = "linux-s390x"
LINUX_PPC64LE = "linux-ppc64le"


LINUX_BINARY_BUILD_WORFKLOWS = [
Expand Down Expand Up @@ -368,6 +369,20 @@ class OperatingSystem:
),
]

PPC64LE_BINARY_BUILD_WORKFLOWS = [
BinaryBuildWorkflow(
os=OperatingSystem.LINUX_PPC64LE,
package_type="manywheel",
build_configs=generate_binary_build_matrix.generate_wheels_matrix(
OperatingSystem.LINUX_PPC64LE
),
ciflow_config=CIFlowConfig(
labels={LABEL_CIFLOW_BINARIES, LABEL_CIFLOW_BINARIES_WHEEL},
isolated_workflow=True,
),
),
]


def main() -> None:
jinja_env = jinja2.Environment(
Expand All @@ -390,6 +405,10 @@ def main() -> None:
jinja_env.get_template("linux_binary_build_workflow.yml.j2"),
S390X_BINARY_BUILD_WORKFLOWS,
),
(
jinja_env.get_template("linux_binary_build_workflow.yml.j2"),
PPC64LE_BINARY_BUILD_WORKFLOWS,
),
(
jinja_env.get_template("linux_binary_build_workflow.yml.j2"),
LINUX_BINARY_SMOKE_WORKFLOWS,
Expand Down
9 changes: 9 additions & 0 deletions .github/templates/linux_binary_build_workflow.yml.j2
Original file line number Diff line number Diff line change
Expand Up @@ -35,6 +35,8 @@ env:
ALPINE_IMAGE: "arm64v8/alpine"
{%- elif "s390x" in build_environment %}
ALPINE_IMAGE: "docker.io/s390x/alpine"
{%- elif "ppc64le" in build_environment %}
ALPINE_IMAGE: "docker.io/ppc64le/alpine"
{%- else %}
ALPINE_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/tool/alpine"
{%- endif %}
Expand Down Expand Up @@ -74,6 +76,10 @@ jobs:
runs_on: linux.s390x
ALPINE_IMAGE: "docker.io/s390x/alpine"
timeout-minutes: 420
{%- elif "ppc64le" in build_environment %}
runs_on: linux.ppc64le
ALPINE_IMAGE: "docker.io/ppc64le/alpine"
timeout-minutes: 420
{%- elif "conda" in build_environment and config["gpu_arch_type"] == "cuda" %}
runner_prefix: "${{ needs.get-label-type.outputs.label-type }}"
runs_on: linux.24xlarge.ephemeral
Expand Down Expand Up @@ -109,6 +115,9 @@ jobs:
{%- elif "s390x" in build_environment %}
runs_on: linux.s390x
ALPINE_IMAGE: "docker.io/s390x/alpine"
{%- elif "ppc64le" in build_environment %}
runs_on: linux.ppc64le
ALPINE_IMAGE: "docker.io/ppc64le/alpine"
{%- elif config["gpu_arch_type"] == "rocm" %}
runs_on: linux.rocm.gpu
{%- elif config["gpu_arch_type"] == "cuda" and config["gpu_arch_version"] == "12.8" %}
Expand Down
Loading
Loading