# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
# SPDX-License-Identifier: Apache-2.0
FROM quay.io/pypa/manylinux2014_x86_64

ARG PYTHON_VERSION=3.9
ARG TENSORFLOW_VERSION=2.9.0
ARG BAZEL_VERSION=5.1.1

RUN ln -s /opt/_internal/cpython-$PYTHON_VERSION*/bin/python3 /usr/local/bin/python3 && \
    ln -s /opt/_internal/cpython-$PYTHON_VERSION*/bin/python /usr/local/bin/python && \
    ln -s /opt/_internal/cpython-$PYTHON_VERSION*/bin/pip3 /usr/local/bin/pip3 && \
    ln -s /opt/_internal/cpython-$PYTHON_VERSION*/bin/pip /usr/local/bin/pip && \
    ln -s /opt/_internal/cpython-$PYTHON_VERSION*/include/python${PYTHON_VERSION}/ /usr/local/include/python

ENV PYTHON_BIN_PATH=/usr/local/bin/python
ENV CI_BUILD_PYTHON=/usr/local/bin/python
ENV CROSSTOOL_PYTHON_INCLUDE_PATH=/usr/local/include/python/

RUN pip install --no-cache-dir setuptools pybind11 numpy twine keyrings.alt GitPython

COPY install/install_bazel.sh /install/
RUN /install/install_bazel.sh ${BAZEL_VERSION}

# CACHE_STOP is used to rerun future commands, otherwise the cloning will be cached and will not pull the most recent version
ARG CACHE_STOP=1
RUN git clone --depth=1 https://github.com/tensorflow/tensorflow.git --branch v${TENSORFLOW_VERSION} /tensorflow_src