Skip to content

Commit 50bbba6

Browse files
authored
cpu-only: inlcude libnvshmem. (#8606)
1 parent 4c8b107 commit 50bbba6

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

build.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1583,6 +1583,7 @@ def add_cpu_libs_to_linux_dockerfile(backends, target_machine):
15831583
COPY --from=min_container /usr/local/cuda/lib64/libcufile.so.0 /usr/local/cuda/targets/{cuda_arch}-linux/lib/.
15841584
COPY --from=min_container /usr/local/cuda/lib64/libnvrtc.so.13 /usr/local/cuda/targets/{cuda_arch}-linux/lib/.
15851585
COPY --from=min_container /usr/local/cuda/lib64/libcusparseLt.so.0 /usr/local/cuda/targets/{cuda_arch}-linux/lib/.
1586+
COPY --from=min_container /usr/local/cuda/lib64/libnvshmem_host.so.3 /usr/local/cuda/targets/{cuda_arch}-linux/lib/.
15861587
15871588
RUN mkdir -p /opt/hpcx/ucc/lib/ /opt/hpcx/ucx/lib/
15881589
COPY --from=min_container /opt/hpcx/ucc/lib/libucc.so.1 /opt/hpcx/ucc/lib/libucc.so.1

0 commit comments

Comments
 (0)