Skip to content

Commit 0c335c4

Browse files
committed
upd tests.yaml files for preset containers
Signed-off-by: jafraustro <jaime.fraustro.valdez@intel.com>
1 parent eca63c5 commit 0c335c4

File tree

2 files changed

+10
-0
lines changed

2 files changed

+10
-0
lines changed

preset/deep-learning/tests.yaml

+4
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ deep-learning-ipex-${PYTHON_VERSION:-3.9}-gpu:
2020
cmd: conda run -n pytorch-gpu python -W ignore sample-tests/intel_extension_for_pytorch/test_ipex.py --device xpu --ipex
2121
img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-deep-learning-${RELEASE:-2024.2.0}-py${PYTHON_VERSION:-3.9}
2222
device: ["/dev/dri"]
23+
user: root
2324

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

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

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

5559
deep-learning-torch-dataset-librarian-${PYTHON_VERSION:-3.9}-cpu:
5660
cmd: conda run -n pytorch-cpu bash -c 'yes | python -m dataset_librarian.dataset -n msmarco --download -d ~/msmarco'

preset/inference-optimization/tests.yaml

+6
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@ inference-optimization-inc-ipex-quantization-notebook-${PYTHON_VERSION:-3.9}-gpu
2222
img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-inference-optimization-${RELEASE:-2024.2.0}-py${PYTHON_VERSION:-3.9}
2323
notebook: True
2424
device: ["/dev/dri"]
25+
user: root
2526

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

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

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

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

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

0 commit comments

Comments
 (0)