Skip to content

Commit 7984974

Browse files
add build parallelism
1 parent f67684c commit 7984974

File tree

4 files changed

+9
-1
lines changed

4 files changed

+9
-1
lines changed

.tekton/odh-training-rocm64-torch29-py312-rhel9-pull-request.yaml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,8 @@ spec:
3333
- name: build-platforms
3434
value:
3535
- linux-d160-m8xlarge/amd64
36+
- name: build-args-file
37+
value: images/universal/training/rocm64-torch290-py312/argfile.konflux.conf
3638
- name: dockerfile
3739
value: Dockerfile
3840
- name: path-context

.tekton/odh-training-rocm64-torch29-py312-rhel9-push.yaml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,8 @@ spec:
3434
value: Dockerfile
3535
- name: path-context
3636
value: images/universal/training/rocm64-torch290-py312
37+
- name: build-args-file
38+
value: images/universal/training/rocm64-torch290-py312/argfile.konflux.conf
3739
pipelineSpec:
3840
timeouts:
3941
pipeline: 40h

images/universal/training/rocm64-torch290-py312/Dockerfile

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -138,7 +138,9 @@ WORKDIR /opt/app-root/src
138138
# 3. Then sync all dependencies with --no-build-isolation
139139
ENV UV_NO_CACHE=1 \
140140
GPU_ARCHS="gfx90a;gfx942" \
141-
PYTORCH_ROCM_ARCH="gfx90a;gfx942"
141+
PYTORCH_ROCM_ARCH="gfx90a;gfx942" \
142+
MAX_JOBS=32 \
143+
CMAKE_BUILD_PARALLEL_LEVEL=32
142144
RUN uv pip install --index-strategy=unsafe-best-match --index-url=https://download.pytorch.org/whl/rocm6.4 --extra-index-url=https://pypi.org/simple "torch==2.9.0+rocm6.4"
143145
RUN uv pip sync --python-platform=linux --python-version=3.12 --no-build-isolation /tmp/deps/pylock.toml
144146
ENV UV_NO_CACHE=
Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
MAX_JOBS=32
2+
CMAKE_BUILD_PARALLEL_LEVEL=32

0 commit comments

Comments
 (0)