diff --git a/Dockerfile b/Dockerfile index f728c9268..4ea9b9ef3 100644 --- a/Dockerfile +++ b/Dockerfile @@ -27,12 +27,13 @@ RUN cargo install cargo-build-deps \ FROM --platform=linux/amd64 build-image as build-app WORKDIR /src/gpu-iris-mpc COPY . . -RUN cargo build --release --target x86_64-unknown-linux-gnu --bin server --bin client --bin key-manager --bin upgrade-server --bin upgrade-client --bin upgrade-checker +RUN cargo build --release --target x86_64-unknown-linux-gnu --bin nccl --bin server --bin client --bin key-manager --bin upgrade-server --bin upgrade-client --bin upgrade-checker FROM --platform=linux/amd64 ghcr.io/worldcoin/iris-mpc-base:cuda12_2-nccl2_22_3_1 ENV DEBIAN_FRONTEND=noninteractive # Include client, server and key-manager, upgrade-client and upgrade-server binaries +COPY --from=build-app /src/gpu-iris-mpc/target/x86_64-unknown-linux-gnu/release/nccl /bin/nccl COPY --from=build-app /src/gpu-iris-mpc/target/x86_64-unknown-linux-gnu/release/server /bin/server COPY --from=build-app /src/gpu-iris-mpc/target/x86_64-unknown-linux-gnu/release/client /bin/client COPY --from=build-app /src/gpu-iris-mpc/target/x86_64-unknown-linux-gnu/release/key-manager /bin/key-manager diff --git a/iris-mpc-gpu/Cargo.toml b/iris-mpc-gpu/Cargo.toml index ab8e6ffe2..813b3da87 100644 --- a/iris-mpc-gpu/Cargo.toml +++ b/iris-mpc-gpu/Cargo.toml @@ -52,3 +52,7 @@ harness = false [[bench]] name = "transpose" harness = false + +[[bin]] +name = "nccl" +path = "src/bin/nccl.rs"