cuquantum-benchmark added

Quleaf · Oct 29, 2024 · 5372f51 · 5372f51
1 parent cb9d046
commit 5372f51
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/...appliance-next/cuquantum-appliance-next-ubuntu/cuquantum-appliance-next-ubuntu.dockerfile b/...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