From db7a79cbf74ec0b21e009faa7f1b563f1f88aeb2 Mon Sep 17 00:00:00 2001 From: Rehan Khan Date: Wed, 29 Jul 2026 10:22:17 +0530 Subject: [PATCH] [CPU] Fix s390x builds and update torch version in dockerfile (#50144) Signed-off-by: Rehan Khan Co-authored-by: Li, Jiang --- csrc/cpu/cpu_types_vxe.hpp | 6 +++--- docker/Dockerfile.s390x | 4 ++-- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/csrc/cpu/cpu_types_vxe.hpp b/csrc/cpu/cpu_types_vxe.hpp index bf96554a8d..26a14e8ea6 100644 --- a/csrc/cpu/cpu_types_vxe.hpp +++ b/csrc/cpu/cpu_types_vxe.hpp @@ -269,7 +269,7 @@ struct FP32Vec4 : public Vec { explicit FP32Vec4(__vector float data) : reg(data) {} - explicit FP32Vec4(const FP32Vec4& data) : reg(data.reg) {} + FP32Vec4(const FP32Vec4& data) : reg(data.reg) {} }; struct FP32Vec8 : public Vec { @@ -298,7 +298,7 @@ struct FP32Vec8 : public Vec { explicit FP32Vec8(f32x4x2_t data) : reg(data) {} - explicit FP32Vec8(const FP32Vec8& data) { + FP32Vec8(const FP32Vec8& data) { reg.val[0] = data.reg.val[0]; reg.val[1] = data.reg.val[1]; } @@ -643,7 +643,7 @@ struct FP32Vec16 : public Vec { explicit FP32Vec16(f32x4x4_t data) : reg(data) {} - explicit FP32Vec16(const FP32Vec16& data) { + FP32Vec16(const FP32Vec16& data) { reg.val[0] = data.reg.val[0]; reg.val[1] = data.reg.val[1]; reg.val[2] = data.reg.val[2]; diff --git a/docker/Dockerfile.s390x b/docker/Dockerfile.s390x index b71e035e15..d645e37e78 100644 --- a/docker/Dockerfile.s390x +++ b/docker/Dockerfile.s390x @@ -61,13 +61,13 @@ ENV C_INCLUDE_PATH="/usr/local/include:$C_INCLUDE_PATH" FROM python-install AS torch-vision # Install torchvision -ARG TORCH_VISION_VERSION=v0.26.0 +ARG TORCH_VISION_VERSION=v0.28.0 WORKDIR /tmp RUN --mount=type=cache,target=/root/.cache/uv \ git clone https://github.com/pytorch/vision.git && \ cd vision && \ git checkout $TORCH_VISION_VERSION && \ - uv pip install torch==2.11.0 --index-url https://download.pytorch.org/whl/cpu && \ + uv pip install torch==2.13.0 --index-url https://download.pytorch.org/whl/cpu && \ python setup.py bdist_wheel FROM python-install AS hf-xet-builder