From 5372f5104725f4deef0e6bb9edfb945977fb7dca Mon Sep 17 00:00:00 2001 From: klinus Date: Tue, 29 Oct 2024 19:36:30 +1100 Subject: [PATCH] cuquantum-benchmark added --- .../cuquantum-appliance-next-ubuntu.dockerfile | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/arm/build-cuquantum-appliance-next/cuquantum-appliance-next-ubuntu/cuquantum-appliance-next-ubuntu.dockerfile b/arm/build-cuquantum-appliance-next/cuquantum-appliance-next-ubuntu/cuquantum-appliance-next-ubuntu.dockerfile index 410011c..a512e07 100644 --- a/arm/build-cuquantum-appliance-next/cuquantum-appliance-next-ubuntu/cuquantum-appliance-next-ubuntu.dockerfile +++ b/arm/build-cuquantum-appliance-next/cuquantum-appliance-next-ubuntu/cuquantum-appliance-next-ubuntu.dockerfile @@ -2,7 +2,7 @@ FROM ubuntu:22.04 LABEL org.opencontainers.image.arch=arm LABEL org.opencontainers.image.compilation=auto -LABEL org.opencontainers.image.devmode=true +LABEL org.opencontainers.image.devmode=false LABEL org.opencontainers.image.ref.name="ubuntu" LABEL org.opencontainers.image.version="22.04" LABEL org.opencontainers.image.author="Shusen Liu" @@ -214,6 +214,12 @@ RUN cd /opt/cuquantum/distributed_interfaces &&\ sh activate_mpi.sh &&\ echo 'export CUTENSORNET_COMM_LIB=/opt/cuquantum/distributed_interfaces/libcutensornet_distributed_mpi.so' >> /opt/cuquantum-env/activate_cuquantum.sh +RUN git clone https://github.com/nvidia/cuquantum.git /tmp/cuquantum && \ + cp -r /tmp/cuquantum/benchmarks /opt/benchmarks && \ + . /opt/cuquantum-env/activate_cuquantum.sh && \ + cd /opt/benchmarks && \ + pip install . + # Configure deactivate script