Skip to content

Commit 9004896

Browse files
committed
Annotate ROCm runtime dependencies
1 parent 5948497 commit 9004896

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

Dockerfile

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -130,8 +130,9 @@ RUN if [ "${BUILD_TYPE}" = "hipblas" ] && [ "${SKIP_DRIVERS}" = "false" ]; then
130130
# End setup steps for specific ROCm version - the packages below will be installed from the configured repositories
131131
apt-get update && \
132132
apt-get install -y --no-install-recommends \
133-
rocblas-dev \
134-
hipblas-dev && \
133+
rocm-hip-runtime # Gives libomp.so\
134+
rocblas-dev # Give librocblas.so \
135+
hipblas-dev # Gives libhipblas.so && \
135136
apt-get clean && \
136137
rm -rf /var/lib/apt/lists/* && \
137138
echo "amd" > /run/localai/capability && \

0 commit comments

Comments
 (0)