|
13 | 13 | # limitations under the License.
|
14 | 14 |
|
15 | 15 | ---
|
16 |
| -# import-itex-cpu-${PACKAGE_OPTION:-pip}: |
17 |
| -# img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-${TF_VERSION:-2.15.1}-base |
18 |
| -# cmd: python -c "from tensorflow.python.client import device_lib; print(device_lib.list_local_devices())" |
19 |
| -# import-itex-xpu-${PACKAGE_OPTION:-pip}: |
20 |
| -# img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-itex-${TF_VERSION:-2.15.1}-itex-xpu-base |
21 |
| -# cmd: python /tests/xpu_import_test.py |
22 |
| -# device: ["/dev/dri"] |
23 |
| -# volumes: |
24 |
| -# - src: ${PWD}/tensorflow/tests |
25 |
| -# dst: /tests |
26 |
| -# import-cpu-jupyter-${PACKAGE_OPTION:-pip}: |
27 |
| -# img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-${TF_VERSION:-2.15.1}-jupyter |
28 |
| -# cmd: python -m jupyter --version |
29 |
| -# import-xpu-jupyter-${PACKAGE_OPTION:-pip}: |
30 |
| -# img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-itex-${TF_VERSION:-2.15.1}-itex-xpu-jupyter |
31 |
| -# cmd: python -m jupyter --version |
32 |
| -# device: ["/dev/dri"] |
33 |
| -# import-multinode-${PACKAGE_OPTION:-pip}: |
34 |
| -# img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-${TF_VERSION:-2.15.1}-horovod-${HOROVOD_VERSION:-0.28.1}-inc-${INC_VERSION:-3.0.2} |
35 |
| -# cmd: horovodrun --check-build && mpirun --version && python -c 'import horovod.tensorflow as hvd;hvd.init();import horovod.tensorflow' |
36 |
| -# import-inc-${PACKAGE_OPTION:-pip}: |
37 |
| -# img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-${TF_VERSION:-2.15.1}-horovod-${HOROVOD_VERSION:-0.28.1}-inc-${INC_VERSION:-3.0.2} |
38 |
| -# cmd: python -c "import neural_compressor as inc;print(inc.__version__)" |
39 |
| -# itex-cpu-${PACKAGE_OPTION:-pip}: |
40 |
| -# img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-${TF_VERSION:-2.15.1}-base |
41 |
| -# cmd: python /tests/tf_base_test.py |
42 |
| -# volumes: |
43 |
| -# - src: ${PWD}/tensorflow/tests |
44 |
| -# dst: /tests |
45 |
| -# itex-xpu-${PACKAGE_OPTION:-pip}: |
46 |
| -# img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-itex-${TF_VERSION:-2.15.1}-itex-xpu-base |
47 |
| -# cmd: python /tests/tf_base_test.py |
48 |
| -# device: ["/dev/dri"] |
49 |
| -# volumes: |
50 |
| -# - dst: /tests |
51 |
| -# src: $PWD/tensorflow/tests |
52 |
| -# itex-xpu-jupyter-${PACKAGE_OPTION:-pip}: |
53 |
| -# img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-itex-${TF_VERSION:-2.15.1}-itex-xpu-jupyter |
54 |
| -# cmd: papermill --log-output /jupyter/xpu.ipynb -k python3 - |
55 |
| -# notebook: True |
56 |
| -# device: ["/dev/dri"] |
57 |
| -# multinode-${PACKAGE_OPTION:-pip}: |
58 |
| -# img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-${TF_VERSION:-2.15.1}-horovod-${HOROVOD_VERSION:-0.28.1}-inc-${INC_VERSION:-3.0.2} |
59 |
| -# cmd: horovodrun -np 2 -H localhost:2 --binding-args="-bind-to socket -map-by socket" python /tests/tf_base_test.py |
60 |
| -# volumes: |
61 |
| -# - dst: /tests |
62 |
| -# src: $PWD/tensorflow/tests |
| 16 | +import-itex-cpu-${PACKAGE_OPTION:-pip}: |
| 17 | + img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-${TF_VERSION:-2.15.1}-base |
| 18 | + cmd: python -c "from tensorflow.python.client import device_lib; print(device_lib.list_local_devices())" |
| 19 | +import-itex-xpu-${PACKAGE_OPTION:-pip}: |
| 20 | + img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-itex-${TF_VERSION:-2.15.1}-itex-xpu-base |
| 21 | + cmd: python /tests/xpu_import_test.py |
| 22 | + device: ["/dev/dri"] |
| 23 | + volumes: |
| 24 | + - src: ${PWD}/tensorflow/tests |
| 25 | + dst: /tests |
| 26 | +import-cpu-jupyter-${PACKAGE_OPTION:-pip}: |
| 27 | + img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-${TF_VERSION:-2.15.1}-jupyter |
| 28 | + cmd: python -m jupyter --version |
| 29 | +import-xpu-jupyter-${PACKAGE_OPTION:-pip}: |
| 30 | + img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-itex-${TF_VERSION:-2.15.1}-itex-xpu-jupyter |
| 31 | + cmd: python -m jupyter --version |
| 32 | + device: ["/dev/dri"] |
| 33 | +import-multinode-${PACKAGE_OPTION:-pip}: |
| 34 | + img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-${TF_VERSION:-2.15.1}-horovod-${HOROVOD_VERSION:-0.28.1}-inc-${INC_VERSION:-3.0.2} |
| 35 | + cmd: horovodrun --check-build && mpirun --version && python -c 'import horovod.tensorflow as hvd;hvd.init();import horovod.tensorflow' |
| 36 | +import-inc-${PACKAGE_OPTION:-pip}: |
| 37 | + img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-${TF_VERSION:-2.15.1}-horovod-${HOROVOD_VERSION:-0.28.1}-inc-${INC_VERSION:-3.0.2} |
| 38 | + cmd: python -c "import neural_compressor as inc;print(inc.__version__)" |
| 39 | +itex-cpu-${PACKAGE_OPTION:-pip}: |
| 40 | + img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-${TF_VERSION:-2.15.1}-base |
| 41 | + cmd: python /tests/tf_base_test.py |
| 42 | + volumes: |
| 43 | + - src: ${PWD}/tensorflow/tests |
| 44 | + dst: /tests |
| 45 | +itex-xpu-${PACKAGE_OPTION:-pip}: |
| 46 | + img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-itex-${TF_VERSION:-2.15.1}-itex-xpu-base |
| 47 | + cmd: python /tests/tf_base_test.py |
| 48 | + device: ["/dev/dri"] |
| 49 | + volumes: |
| 50 | + - dst: /tests |
| 51 | + src: $PWD/tensorflow/tests |
| 52 | +itex-xpu-jupyter-${PACKAGE_OPTION:-pip}: |
| 53 | + img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-itex-${TF_VERSION:-2.15.1}-itex-xpu-jupyter |
| 54 | + cmd: papermill --log-output /jupyter/xpu.ipynb -k python3 - |
| 55 | + notebook: True |
| 56 | + device: ["/dev/dri"] |
| 57 | +multinode-${PACKAGE_OPTION:-pip}: |
| 58 | + img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-${TF_VERSION:-2.15.1}-horovod-${HOROVOD_VERSION:-0.28.1}-inc-${INC_VERSION:-3.0.2} |
| 59 | + cmd: horovodrun -np 2 -H localhost:2 --binding-args="-bind-to socket -map-by socket" python /tests/tf_base_test.py |
| 60 | + volumes: |
| 61 | + - dst: /tests |
| 62 | + src: $PWD/tensorflow/tests |
63 | 63 | inc-${PACKAGE_OPTION:-pip}:
|
64 | 64 | img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-${TF_VERSION:-2.15.1}-horovod-${HOROVOD_VERSION:-0.28.1}-inc-${INC_VERSION:-3.0.2}
|
65 | 65 | cmd: bash /tests/inc_test.sh
|
|
0 commit comments