@@ -22,6 +22,7 @@ inference-optimization-inc-ipex-quantization-notebook-${PYTHON_VERSION:-3.9}-gpu
22
22
img : ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-inference-optimization-${RELEASE:-2024.2.0}-py${PYTHON_VERSION:-3.9}
23
23
notebook : True
24
24
device : ["/dev/dri"]
25
+ user : root
25
26
26
27
inference-optimization-inc-itex-notebook-${PYTHON_VERSION:-3.9}-cpu :
27
28
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:
53
54
cmd : conda run -n pytorch-gpu python -W ignore sample-tests/intel_extension_for_pytorch/test_ipex.py --device xpu --ipex
54
55
img : ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-inference-optimization-${RELEASE:-2024.2.0}-py${PYTHON_VERSION:-3.9}
55
56
device : ["/dev/dri"]
57
+ user : root
56
58
57
59
inference-optimization-itex-${PYTHON_VERSION:-3.9}-cpu :
58
60
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:
61
63
cmd : conda run -n tensorflow-gpu python -W ignore sample-tests/intel_extension_for_tensorflow/test_itex.py
62
64
img : ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-inference-optimization-${RELEASE:-2024.2.0}-py${PYTHON_VERSION:-3.9}
63
65
device : ["/dev/dri"]
66
+ user : root
64
67
65
68
inference-optimization-itex-inference-notebook-${PYTHON_VERSION:-3.9}-cpu :
66
69
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:
80
83
cmd : conda run -n tensorflow-gpu sample-tests/onnx/run.sh
81
84
img : ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-inference-optimization-${RELEASE:-2024.2.0}-py${PYTHON_VERSION:-3.9}
82
85
device : ["/dev/dri"]
86
+ user : root
83
87
84
88
inference-optimization-tensorflow-dataset-librarian-${PYTHON_VERSION:-3.9}-cpu :
85
89
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:
88
92
cmd : conda run -n tensorflow-gpu bash -c 'yes | python -m dataset_librarian.dataset -n msmarco --download -d ~/msmarco'
89
93
img : ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-inference-optimization-${RELEASE:-2024.2.0}-py${PYTHON_VERSION:-3.9}
90
94
device : ["/dev/dri"]
95
+ user : root
91
96
92
97
inference-optimization-torch-dataset-librarian-${PYTHON_VERSION:-3.9}-cpu :
93
98
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:
96
101
cmd : conda run -n pytorch-gpu bash -c 'yes | python -m dataset_librarian.dataset -n msmarco --download -d ~/msmarco'
97
102
img : ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-inference-optimization-${RELEASE:-2024.2.0}-py${PYTHON_VERSION:-3.9}
98
103
device : ["/dev/dri"]
104
+ user : root
0 commit comments