Dockerfile.stage4

ARG TARGETPLATFORM
ARG FROMPLATFORM=${TARGETPLATFORM}
ARG FROMIMAGE
FROM --platform=${FROMPLATFORM} ${FROMIMAGE}

COPY arm.patch /src/torch/arm.patch
RUN \
  cd pytorch && \
  timeout --preserve-status --kill-after=330m --verbose 300m python setup.py build || :