final first commit

qibojit-benchmarks/scripts/multigpu.sh

@@ -0,0 +1,34 @@
+# Generates the multigpu bar plot
+# Assumes a machine with four GPUs
+
+: "${filename:=multigpu.dat}"
+: "${precision:=double}"
+: "${nreps:=1}"
+
+export CUDA_VISIBLE_DEVICES=0,1,2,3
+for circuit in qft variational supremacy qv bv
+do
+  for backend in qibojit qibotf
+  do
+    python compare.py --circuit $circuit --nqubits 32 --filename $filename \
+                      --library-options backend=$backend,accelerators=1/GPU:0+1/GPU:1+1/GPU:2+1/GPU:3 \
+                      --nreps $nreps --precision $precision
+    echo
+  done
+
+  for backend in qibojit qibotf
+  do
+    python compare.py --circuit $circuit --nqubits 32 --filename $filename \
+                      --library-options backend=$backend,accelerators=2/GPU:2+2/GPU:3 \
+                      --nreps $nreps --precision $precision
+    echo
+  done
+
+  for backend in qibojit qibotf
+  do
+    python compare.py --circuit $circuit --nqubits 32 --filename $filename \
+                      --library-options backend=$backend,accelerators=4/GPU:3 \
+                      --nreps $nreps --precision $precision
+    echo
+  done
+done