final first commit

qibojit-benchmarks/scripts/qibotn_local.sh

@@ -0,0 +1,56 @@
+#!/usr/bin/env bash
+# Lightweight local benchmark script (single node, small sweep)
+set -euo pipefail
+
+: "${circuit:=variational}"
+: "${precision:=complex128}"
+: "${nreps:=1}"
+: "${filename:=qibotn_expectation_local.dat}"
+: "${np:=2}"
+: "${nlayers:=2}"
+: "${nqubits_list:=8 12 16}"
+
+script_dir="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+repo_root="$(cd "${script_dir}/.." && pwd)"
+cd "${repo_root}"
+
+# On Prism, Intel MPI may need shared-memory fabric for local runs.
+if [[ -z "${I_MPI_FABRICS:-}" ]]; then
+    export I_MPI_FABRICS=shm
+fi
+
+for nqubits in ${nqubits_list}; do
+    echo "===== nqubits=${nqubits} ====="
+
+    # MPI-enabled cutensornet expectation
+    mpirun -np "${np}" python compare.py \
+        --circuit "${circuit}" \
+        --circuit-options "nlayers=${nlayers}" \
+        --nqubits "${nqubits}" \
+        --filename "${filename}" \
+        --library-options backend=qibotn,platform=cutensornet,computation_settings=scripts/cu_tensornet_mpi_expectation.json \
+        --nreps "${nreps}" \
+        --precision "${precision}"
+
+    # Single-process cutensornet expectation
+    python compare.py \
+        --circuit "${circuit}" \
+        --circuit-options "nlayers=${nlayers}" \
+        --nqubits "${nqubits}" \
+        --filename "${filename}" \
+        --library-options backend=qibotn,platform=cutensornet,computation_settings=scripts/cu_tensornet_expectation.json \
+        --nreps "${nreps}" \
+        --precision "${precision}"
+
+    # qibojit reference
+    python compare.py \
+        --circuit "${circuit}" \
+        --circuit-options "nlayers=${nlayers}" \
+        --nqubits "${nqubits}" \
+        --filename "${filename}" \
+        --library-options backend=qibojit,platform=numba,expectation=XXXZ \
+        --nreps "${nreps}" \
+        --precision "${precision}"
+
+    echo
+ done