diff --git a/Dockerfile b/Dockerfile index 12e8139c9..1e356014d 100644 --- a/Dockerfile +++ b/Dockerfile @@ -100,6 +100,10 @@ RUN if [ "${BUILD_TYPE}" = "hipblas" ] && [ "${SKIP_DRIVERS}" = "false" ]; then ldconfig \ ; fi +RUN if [ "${BUILD_TYPE}" = "hipblas" ]; then \ + ln -s /opt/rocm-**/lib/llvm/lib/libomp.so /usr/lib/libomp.so \ + ; fi + RUN expr "${BUILD_TYPE}" = intel && echo "intel" > /run/localai/capability || echo "not intel" # Cuda