chore(docker): multi arch support

Signed-off-by: Wataru Ishida <wataru.ishid@gmail.com>
osrg · Mar 12, 2024 · f6e89a9 · f6e89a9
1 parent fef67dc
commit f6e89a9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -8,7 +8,9 @@ COPY test/nccl test/nccl
 RUN cd test && make build-nccl
 
 COPY test/nccl-tests test/nccl-tests
-RUN cd test && MPI_HOME=/usr/lib/x86_64-linux-gnu/openmpi make build-nccl-tests
+COPY --from=tonistiigi/xx / /
+ARG TARGETPLATFORM
+RUN cd test && MPI_HOME=/usr/lib/$(xx-info march)-linux-gnu/openmpi make build-nccl-tests
 
 COPY nccl_plugin nccl_plugin