[ROCm] Ship rocprofiler-sdk 1.3.2 in Dockerfile.rocm_base to fix torch.profiler traces (#53182)
Signed-off-by: Rohan138 <rohanpotdar138@gmail.com>
Rohan Potdar committed
Aug 22, 2026 at 12:16 UTC
610bfc58e93c652fc8a075bc814ba1617f5fcf2b
1 file changed
+41
docker/Dockerfile.rocm_base
+41
@@ -13,6 +13,8 @@ ARG AITER_BRANCH="v0.1.19"
13
ARG AITER_REPO="https://github.com/ROCm/aiter.git"
14
ARG MORI_BRANCH="v1.1.0"
15
ARG MORI_REPO="https://github.com/ROCm/mori.git"
16
+ARG ROCPROFILER_SDK_REPO="https://github.com/ROCm/rocm-systems.git"
17
+ARG ROCPROFILER_SDK_COMMIT="2b22ab0195cc1461cd9abf3b969e9dd7c10af350"
18
19
# Sccache configuration (only used in release pipeline)
20
ARG USE_SCCACHE
@@ -313,10 +315,45 @@ RUN --mount=type=bind,from=build_aiter,src=/app/install/,target=/install \
315
RUN --mount=type=bind,from=build_mori,src=/app/install/,target=/install \
316
cp /install/*.whl /app/debs
317
318
+# rocprofiler-sdk 1.3.2 fixes torch 2.12 profiler on ROCm 7.2.3
319
+FROM base AS build_rocprofiler_sdk
320
+ARG ROCPROFILER_SDK_REPO
321
+ARG ROCPROFILER_SDK_COMMIT
322
+ARG USE_SCCACHE
323
+RUN apt-get update \
324
+ && apt-get install -y --no-install-recommends libdw-dev libelf-dev libsqlite3-dev libzstd-dev \
325
+ && rm -rf /var/lib/apt/lists/*
326
+RUN git init -q /src && cd /src \
327
+ && git remote add origin ${ROCPROFILER_SDK_REPO} \
328
+ && git config core.sparseCheckout true \
329
+ && git config remote.origin.promisor true \
330
+ && git config remote.origin.partialclonefilter blob:none \
331
+ && git sparse-checkout init --cone \
332
+ && git sparse-checkout set projects/rocprofiler-sdk shared cmake \
333
+ && git fetch --filter=blob:none --depth 1 origin ${ROCPROFILER_SDK_COMMIT} \
334
+ && git checkout -q FETCH_HEAD
335
+RUN cd /src/projects/rocprofiler-sdk \
336
+ && if [ "$USE_SCCACHE" = "1" ]; then \
337
+ export CMAKE_C_COMPILER_LAUNCHER=sccache \
338
+ && export CMAKE_CXX_COMPILER_LAUNCHER=sccache; \
339
+ fi \
340
+ && cmake -B build -G Ninja \
341
+ -DCMAKE_BUILD_TYPE=Release \
342
+ -DCMAKE_PREFIX_PATH=/opt/rocm \
343
+ -DCMAKE_INSTALL_PREFIX=/opt/rocm \
344
+ -DROCPROFILER_BUILD_TESTS=OFF \
345
+ -DROCPROFILER_BUILD_SAMPLES=OFF \
346
+ -DROCPROFILER_BUILD_DOCS=OFF \
347
+ && cmake --build build --parallel "$(nproc)" \
348
+ && cmake --install build --prefix /staging --strip
349
+
350
FROM base AS final
351
RUN --mount=type=bind,from=debs,src=/app/debs,target=/install \
352
pip install /install/*.whl
353
354
+COPY --from=build_rocprofiler_sdk /staging/ /opt/rocm/
355
+RUN ldconfig
356
+
357
ARG BASE_IMAGE
358
ARG TRITON_BRANCH
359
ARG TRITON_REPO
@@ -332,7 +369,11 @@ ARG AITER_BRANCH
369
ARG AITER_REPO
370
ARG MORI_BRANCH
371
ARG MORI_REPO
372
+ARG ROCPROFILER_SDK_REPO
373
+ARG ROCPROFILER_SDK_COMMIT
374
RUN echo "BASE_IMAGE: ${BASE_IMAGE}" > /app/versions.txt \
375
+ && echo "ROCPROFILER_SDK_REPO: ${ROCPROFILER_SDK_REPO}" >> /app/versions.txt \
376
+ && echo "ROCPROFILER_SDK_COMMIT: ${ROCPROFILER_SDK_COMMIT}" >> /app/versions.txt \
377
&& echo "TRITON_BRANCH: ${TRITON_BRANCH}" >> /app/versions.txt \
378
&& echo "TRITON_REPO: ${TRITON_REPO}" >> /app/versions.txt \
379
&& echo "PYTORCH_BRANCH: ${PYTORCH_BRANCH}" >> /app/versions.txt \