File tree 10 files changed +20
-65
lines changed
10 files changed +20
-65
lines changed Original file line number Diff line number Diff line change 13
13
- requirements.txt
14
14
- .azure-pipelines/code-scan.yml
15
15
- .azure-pipelines/scripts/codeScan
16
+ - .azure-pipelines/template/docker-template.yml
16
17
17
18
pool :
18
19
vmImage : " ubuntu-latest"
Load Diff This file was deleted.
Original file line number Diff line number Diff line change 15
15
- requirements_pt.txt
16
16
- .azure-pipelines/scripts/models
17
17
- .azure-pipelines/model-test-3x.yml
18
+ - .azure-pipelines/template/docker-template.yml
18
19
19
20
variables :
20
21
OUT_SCRIPT_PATH : $(Build.SourcesDirectory)/.azure-pipelines/scripts/models
Original file line number Diff line number Diff line change 12
12
- setup.py
13
13
- requirements.txt
14
14
- .azure-pipelines/model-test.yml
15
+ - .azure-pipelines/template/docker-template.yml
15
16
- .azure-pipelines/scripts/models
16
17
- examples/tensorflow/oob_models/quantization/ptq
17
18
- .azure-pipelines/model-test.yml
Original file line number Diff line number Diff line change 74
74
75
75
- ${{ if eq(parameters.imageSource, 'pull') }} :
76
76
- script : |
77
- docker pull vault.habana.ai/gaudi-docker/1.19.0/ubuntu24 .04/habanalabs/pytorch-installer-2.5.1:latest
77
+ docker pull vault.habana.ai/gaudi-docker/1.19.0/ubuntu22 .04/habanalabs/pytorch-installer-2.5.1:latest
78
78
displayName: "Pull habana docker image"
79
79
80
80
- script : |
95
95
else
96
96
docker run -dit --disable-content-trust --privileged --name=${{ parameters.containerName }} --shm-size="2g" \
97
97
--runtime=habana -e HABANA_VISIBLE_DEVICES=all -e OMPI_MCA_btl_vader_single_copy_mechanism=none --cap-add=sys_nice --net=host --ipc=host \
98
- -v ${BUILD_SOURCESDIRECTORY}:/neural-compressor vault.habana.ai/gaudi-docker/1.19.0/ubuntu24.04/habanalabs/pytorch-installer-2.5.1:latest
98
+ -v ${BUILD_SOURCESDIRECTORY}:/neural-compressor vault.habana.ai/gaudi-docker/1.19.0/ubuntu22.04/habanalabs/pytorch-installer-2.5.1:latest
99
+ docker exec ${{ parameters.containerName }} bash -c "ln -sf \$(which python3) /usr/bin/python"
99
100
fi
100
101
echo "Show the container list after docker run ... "
101
102
docker ps -a
Original file line number Diff line number Diff line change 11
11
- .azure-pipelines/scripts/ut/3x/run_3x_pt_fp8.sh
12
12
- .azure-pipelines/scripts/install_nc.sh
13
13
- .azure-pipelines/ut-3x-pt-fp8.yml
14
+ - .azure-pipelines/template/docker-template.yml
14
15
- neural_compressor/common
15
16
- neural_compressor/torch
16
17
- test/3x/torch/algorithms/fp8_quant
@@ -88,7 +89,7 @@ stages:
88
89
89
90
- script : |
90
91
echo "--- create container ---"
91
- docker run -d -it --name="collectLogs" -v ${BUILD_SOURCESDIRECTORY}:/neural-compressor ${IMAGE_NAME}:${IMAGE_TAG} /bin/bash
92
+ docker run -d -it --name="collectLogs" -v ${BUILD_SOURCESDIRECTORY}:/neural-compressor ${IMAGE_NAME}:${IMAGE_TAG} /bin/bash
92
93
echo "--- docker ps ---"
93
94
docker ps
94
95
echo "--- collect logs ---"
Original file line number Diff line number Diff line change 15
15
- setup.py
16
16
- requirements_pt.txt
17
17
- .azure-pipelines/ut-3x-pt.yml
18
+ - .azure-pipelines/template/docker-template.yml
18
19
- .azure-pipelines/scripts/install_nc.sh
19
20
- .azure-pipelines/scripts/ut/3x/run_3x_pt.sh
20
21
@@ -86,7 +87,7 @@ stages:
86
87
87
88
- script : |
88
89
echo "--- create container ---"
89
- docker run -d -it --name="collectLogs" -v ${BUILD_SOURCESDIRECTORY}:/neural-compressor ${IMAGE_NAME}:${IMAGE_TAG} /bin/bash
90
+ docker run -d -it --name="collectLogs" -v ${BUILD_SOURCESDIRECTORY}:/neural-compressor ${IMAGE_NAME}:${IMAGE_TAG} /bin/bash
90
91
echo "--- docker ps ---"
91
92
docker ps
92
93
echo "--- collect logs ---"
Original file line number Diff line number Diff line change 15
15
- setup.py
16
16
- requirements_tf.txt
17
17
- .azure-pipelines/scripts/ut/3x/run_3x_tf.sh
18
+ - .azure-pipelines/template/docker-template.yml
18
19
19
20
pool : ICX-16C
20
21
@@ -83,7 +84,7 @@ stages:
83
84
84
85
- script : |
85
86
echo "--- create container ---"
86
- docker run -d -it --name="collectLogs" -v ${BUILD_SOURCESDIRECTORY}:/neural-compressor ${IMAGE_NAME}:${IMAGE_TAG} /bin/bash
87
+ docker run -d -it --name="collectLogs" -v ${BUILD_SOURCESDIRECTORY}:/neural-compressor ${IMAGE_NAME}:${IMAGE_TAG} /bin/bash
87
88
echo "--- docker ps ---"
88
89
docker ps
89
90
echo "--- collect logs ---"
Original file line number Diff line number Diff line change 13
13
- setup.py
14
14
- requirements.txt
15
15
- .azure-pipelines/ut-basic.yml
16
+ - .azure-pipelines/template/docker-template.yml
16
17
- .azure-pipelines/scripts/ut
17
18
- .azure-pipelines/scripts/fwk_version.sh
18
19
- .azure-pipelines/scripts/install_nc.sh
@@ -257,7 +258,7 @@ stages:
257
258
258
259
- script : |
259
260
echo "--- create container ---"
260
- docker run -d -it --name="collectLogs" -v ${BUILD_SOURCESDIRECTORY}:/neural-compressor ${IMAGE_NAME}:${IMAGE_TAG} /bin/bash
261
+ docker run -d -it --name="collectLogs" -v ${BUILD_SOURCESDIRECTORY}:/neural-compressor ${IMAGE_NAME}:${IMAGE_TAG} /bin/bash
261
262
echo "--- docker ps ---"
262
263
docker ps
263
264
echo "--- collect logs ---"
Original file line number Diff line number Diff line change @@ -5,8 +5,8 @@ Please note that `INC_VER` must be set to a valid version published here:
5
5
https://pypi.org/project/neural-compressor/#history
6
6
7
7
``` console
8
- $ PYTHON=python3.8
9
- $ INC_VER=1.12
8
+ $ PYTHON=python3.10
9
+ $ INC_BRANCH=3.2
10
10
$ IMAGE_NAME=neural-compressor
11
11
$ IMAGE_TAG=${INC_VER}
12
12
$ docker build --build-arg PYTHON=${PYTHON} --build-arg INC_VER=${INC_VER} -f Dockerfile -t ${IMAGE_NAME} :${IMAGE_TAG} .
@@ -19,8 +19,8 @@ https://docs.docker.com/engine/reference/commandline/tag/
19
19
then please modify the tag so that the tagging requirement is met. For example replace ` / ` with ` - ` .
20
20
21
21
``` console
22
- $ PYTHON=python3.8
23
- $ INC_BRANCH=v1.12
22
+ $ PYTHON=python3.10
23
+ $ INC_BRANCH=3.2
24
24
$ IMAGE_NAME=neural-compressor
25
25
$ IMAGE_TAG=${INC_BRANCH} -devel
26
26
$ docker build --build-arg PYTHON=${PYTHON} --build-arg INC_BRANCH=${INC_BRANCH} -f Dockerfile.devel -t ${IMAGE_NAME} :${IMAGE_TAG} .
@@ -29,6 +29,6 @@ $ docker build --build-arg PYTHON=${PYTHON} --build-arg INC_BRANCH=${INC_BRANCH}
29
29
### Check the Containers built:
30
30
``` console
31
31
$ docker images | grep -i neural-compressor
32
- neural-compressor v1.12 -devel 5c0dc1371312 5 minutes ago 2.76GB
33
- neural-compressor 1.12 303de7f7c38d 36 minutes ago 1.61GB
32
+ neural-compressor v3.2 -devel 5c0dc1371312 5 minutes ago 2.76GB
33
+ neural-compressor 3.2 303de7f7c38d 36 minutes ago 1.61GB
34
34
```
You can’t perform that action at this time.
0 commit comments