Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add the user key to the set of tests in the test.yaml file #476

Merged
merged 9 commits into from
Nov 4, 2024
3 changes: 2 additions & 1 deletion preset/deep-learning/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -103,7 +103,8 @@ RUN apt-get install -y --no-install-recommends --fix-missing \
rm -rf /var/lib/apt/lists/* /etc/apt/sources.list.d/*list

RUN useradd --uid 1000 -d /home/dev -s /bin/bash dev
RUN groupadd -g 109 render
ARG GID=109
RUN groupadd -g ${GID} render
RUN usermod -aG root,sudo,video,render dev
RUN echo 'dev:password' | chpasswd
USER dev
Expand Down
4 changes: 4 additions & 0 deletions preset/deep-learning/tests.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@ deep-learning-ipex-${PYTHON_VERSION:-3.9}-gpu:
cmd: conda run -n pytorch-gpu python -W ignore sample-tests/intel_extension_for_pytorch/test_ipex.py --device xpu --ipex
img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-deep-learning-${RELEASE:-2024.2.0}-py${PYTHON_VERSION:-3.9}
device: ["/dev/dri"]
user: root

deep-learning-ipex-notebook-${PYTHON_VERSION:-3.9}-cpu:
cmd: papermill --log-output jupyter/ipex/ResNet50_Inference.ipynb -k pytorch-cpu
Expand All @@ -30,6 +31,7 @@ deep-learning-ipex-notebook-${PYTHON_VERSION:-3.9}-gpu:
img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-deep-learning-${RELEASE:-2024.2.0}-py${PYTHON_VERSION:-3.9}
notebook: True
device: ["/dev/dri"]
user: root

deep-learning-ipex-quantization-notebook-${PYTHON_VERSION:-3.9}-cpu:
cmd: papermill --log-output jupyter/ipex-quantization/IntelPytorch_Quantization.ipynb -k pytorch-cpu
Expand All @@ -43,6 +45,7 @@ deep-learning-itex-${PYTHON_VERSION:-3.9}-gpu:
cmd: conda run -n tensorflow-gpu python -W ignore sample-tests/intel_extension_for_tensorflow/test_itex.py
img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-deep-learning-${RELEASE:-2024.2.0}-py${PYTHON_VERSION:-3.9}
device: ["/dev/dri"]
user: root

deep-learning-tensorflow-dataset-librarian-${PYTHON_VERSION:-3.9}-cpu:
cmd: conda run -n tensorflow-cpu bash -c 'yes | python -m dataset_librarian.dataset -n msmarco --download -d ~/msmarco'
Expand All @@ -51,6 +54,7 @@ deep-learning-tensorflow-dataset-librarian-${PYTHON_VERSION:-3.9}-gpu:
cmd: conda run -n tensorflow-gpu bash -c 'yes | python -m dataset_librarian.dataset -n msmarco --download -d ~/msmarco'
img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-deep-learning-${RELEASE:-2024.2.0}-py${PYTHON_VERSION:-3.9}
device: ["/dev/dri"]
user: root

deep-learning-torch-dataset-librarian-${PYTHON_VERSION:-3.9}-cpu:
cmd: conda run -n pytorch-cpu bash -c 'yes | python -m dataset_librarian.dataset -n msmarco --download -d ~/msmarco'
Expand Down
6 changes: 6 additions & 0 deletions preset/inference-optimization/tests.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,7 @@ inference-optimization-inc-ipex-quantization-notebook-${PYTHON_VERSION:-3.9}-gpu
img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-inference-optimization-${RELEASE:-2024.2.0}-py${PYTHON_VERSION:-3.9}
notebook: True
device: ["/dev/dri"]
user: root

inference-optimization-inc-itex-notebook-${PYTHON_VERSION:-3.9}-cpu:
cmd: papermill --log-output jupyter/inc-itex/inc_sample_tensorflow.ipynb result.ipynb -k tensorflow-cpu --cwd jupyter/inc-itex
Expand Down Expand Up @@ -53,6 +54,7 @@ inference-optimization-ipex-${PYTHON_VERSION:-3.9}-gpu:
cmd: conda run -n pytorch-gpu python -W ignore sample-tests/intel_extension_for_pytorch/test_ipex.py --device xpu --ipex
img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-inference-optimization-${RELEASE:-2024.2.0}-py${PYTHON_VERSION:-3.9}
device: ["/dev/dri"]
user: root

inference-optimization-itex-${PYTHON_VERSION:-3.9}-cpu:
cmd: conda run -n tensorflow-cpu python -W ignore sample-tests/intel_extension_for_tensorflow/test_itex.py
Expand All @@ -61,6 +63,7 @@ inference-optimization-itex-${PYTHON_VERSION:-3.9}-gpu:
cmd: conda run -n tensorflow-gpu python -W ignore sample-tests/intel_extension_for_tensorflow/test_itex.py
img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-inference-optimization-${RELEASE:-2024.2.0}-py${PYTHON_VERSION:-3.9}
device: ["/dev/dri"]
user: root

inference-optimization-itex-inference-notebook-${PYTHON_VERSION:-3.9}-cpu:
cmd: papermill --log-output jupyter/itex-inference/tutorial_optimize_TensorFlow_pretrained_model.ipynb result.ipynb -k tensorflow-cpu --cwd jupyter/itex-inference
Expand All @@ -80,6 +83,7 @@ inference-optimization-onnx-${PYTHON_VERSION:-3.9}-gpu:
cmd: conda run -n tensorflow-gpu sample-tests/onnx/run.sh
img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-inference-optimization-${RELEASE:-2024.2.0}-py${PYTHON_VERSION:-3.9}
device: ["/dev/dri"]
user: root

inference-optimization-tensorflow-dataset-librarian-${PYTHON_VERSION:-3.9}-cpu:
cmd: conda run -n tensorflow-cpu bash -c 'yes | python -m dataset_librarian.dataset -n msmarco --download -d ~/msmarco'
Expand All @@ -88,6 +92,7 @@ inference-optimization-tensorflow-dataset-librarian-${PYTHON_VERSION:-3.9}-gpu:
cmd: conda run -n tensorflow-gpu bash -c 'yes | python -m dataset_librarian.dataset -n msmarco --download -d ~/msmarco'
img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-inference-optimization-${RELEASE:-2024.2.0}-py${PYTHON_VERSION:-3.9}
device: ["/dev/dri"]
user: root

inference-optimization-torch-dataset-librarian-${PYTHON_VERSION:-3.9}-cpu:
cmd: conda run -n pytorch-cpu bash -c 'yes | python -m dataset_librarian.dataset -n msmarco --download -d ~/msmarco'
Expand All @@ -96,3 +101,4 @@ inference-optimization-torch-dataset-librarian-${PYTHON_VERSION:-3.9}-gpu:
cmd: conda run -n pytorch-gpu bash -c 'yes | python -m dataset_librarian.dataset -n msmarco --download -d ~/msmarco'
img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-inference-optimization-${RELEASE:-2024.2.0}-py${PYTHON_VERSION:-3.9}
device: ["/dev/dri"]
user: root
Loading