[CPU] Fix s390x builds and update torch version in dockerfile (#50144)
Signed-off-by: Rehan Khan <Rehan.Khan7@ibm.com> Co-authored-by: Li, Jiang <jiang1.li@intel.com>
Rehan Khan committed
Jul 29, 2026 at 10:22 UTC
db7a79cbf74ec0b21e009faa7f1b563f1f88aeb2
2 files changed
+5
-5
csrc/cpu/cpu_types_vxe.hpp
+3
-3
@@ -269,7 +269,7 @@ struct FP32Vec4 : public Vec<FP32Vec4> {
269
270
explicit FP32Vec4(__vector float data) : reg(data) {}
271
272
- explicit FP32Vec4(const FP32Vec4& data) : reg(data.reg) {}
272
+ FP32Vec4(const FP32Vec4& data) : reg(data.reg) {}
273
};
274
275
struct FP32Vec8 : public Vec<FP32Vec8> {
@@ -298,7 +298,7 @@ struct FP32Vec8 : public Vec<FP32Vec8> {
298
299
explicit FP32Vec8(f32x4x2_t data) : reg(data) {}
300
301
- explicit FP32Vec8(const FP32Vec8& data) {
301
+ FP32Vec8(const FP32Vec8& data) {
302
reg.val[0] = data.reg.val[0];
303
reg.val[1] = data.reg.val[1];
304
}
@@ -643,7 +643,7 @@ struct FP32Vec16 : public Vec<FP32Vec16> {
643
644
explicit FP32Vec16(f32x4x4_t data) : reg(data) {}
645
646
- explicit FP32Vec16(const FP32Vec16& data) {
646
+ FP32Vec16(const FP32Vec16& data) {
647
reg.val[0] = data.reg.val[0];
648
reg.val[1] = data.reg.val[1];
649
reg.val[2] = data.reg.val[2];
docker/Dockerfile.s390x
+2
-2
@@ -61,13 +61,13 @@ ENV C_INCLUDE_PATH="/usr/local/include:$C_INCLUDE_PATH"
61
62
FROM python-install AS torch-vision
63
# Install torchvision
64
-ARG TORCH_VISION_VERSION=v0.26.0
64
+ARG TORCH_VISION_VERSION=v0.28.0
65
WORKDIR /tmp
66
RUN --mount=type=cache,target=/root/.cache/uv \
67
git clone https://github.com/pytorch/vision.git && \
68
cd vision && \
69
git checkout $TORCH_VISION_VERSION && \
70
- uv pip install torch==2.11.0 --index-url https://download.pytorch.org/whl/cpu && \
70
+ uv pip install torch==2.13.0 --index-url https://download.pytorch.org/whl/cpu && \
71
python setup.py bdist_wheel
72
73
FROM python-install AS hf-xet-builder