Skip to content

Commit

Permalink
Install g++-11 for OneDPL TBB backend
Browse files Browse the repository at this point in the history
  • Loading branch information
ssheorey committed Dec 30, 2024
1 parent 973670e commit 061059d
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 2 deletions.
4 changes: 4 additions & 0 deletions docker/Dockerfile.ci
Original file line number Diff line number Diff line change
Expand Up @@ -87,7 +87,11 @@ RUN apt-get update && apt-get install -y \
libxmlsec1-dev \
libffi-dev \
liblzma-dev \
&& if [ "${BUILD_SYCL_MODULE}" = "ON" ]; then \
apt-get install g++-11; \
fi \
&& rm -rf /var/lib/apt/lists/*
# OneDPL TBB backend requires libstdc++ >= v11

# pyenv
# The pyenv python paths are used during docker run, in this way docker run
Expand Down
5 changes: 3 additions & 2 deletions util/ci_utils.sh
Original file line number Diff line number Diff line change
Expand Up @@ -80,10 +80,11 @@ install_python_dependencies() {
fi
if [ "$BUILD_PYTORCH_OPS" == "ON" ]; then # ML/requirements-torch.txt
if [[ "$OSTYPE" == "linux-gnu"* && "$BUILD_SYCL_MODULE" == "OFF" ]]; then
python -m pip install -U "${TORCH_GLNX}" -f "$TORCH_REPO_URL" tensorboard
python -m pip install -U "${TORCH_GLNX}" -f "$TORCH_REPO_URL"
python -m pip install tensorboard
elif [[ "$OSTYPE" == "linux-gnu"* && "$BUILD_SYCL_MODULE" == "ON" ]]; then
python -m pip install -U "${TORCH_GLNX}.cxx11.abi" -i "$TORCH_CXX11_URL"
python -m pip install -U tensorboard
python -m pip install tensorboard
elif [[ "$OSTYPE" == "darwin"* ]]; then
python -m pip install -U torch=="$TORCH_VER" -f "$TORCH_REPO_URL" tensorboard
else
Expand Down

0 comments on commit 061059d

Please sign in to comment.