Files
qibo-final/qibojit-benchmarks/scripts/multigpu.sh
2026-05-19 17:19:36 +08:00

34 lines
1.1 KiB
Bash
Executable File

# 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