jaunatisblue
4f2dde3464
Some checks failed
Build wheels / build (ubuntu-latest, 3.11) (push) Has been cancelled
Build wheels / build (ubuntu-latest, 3.12) (push) Has been cancelled
Build wheels / build (ubuntu-latest, 3.13) (push) Has been cancelled
Tests / check (push) Has been cancelled
Tests / build (ubuntu-latest, 3.11) (push) Has been cancelled
Tests / build (ubuntu-latest, 3.12) (push) Has been cancelled
Tests / build (ubuntu-latest, 3.13) (push) Has been cancelled
77 lines
1.6 KiB
Markdown
77 lines
1.6 KiB
Markdown
# TN
|
|
```bash
|
|
# qibotn目录下
|
|
I_MPI_FABRICS=shm:ofi \
|
|
I_MPI_OFI_PROVIDER=tcp \
|
|
FI_PROVIDER=tcp \
|
|
CASE=main1 \
|
|
OBSERVABLES=long_z_string \
|
|
NQUBITS=34 \
|
|
NLAYERS=20 \
|
|
TORCH_THREADS=48 \
|
|
SEARCH_REPEATS=2048 \
|
|
SEARCH_TIME=300 \
|
|
SCHEDULER_HOST=10.20.1.103 \
|
|
WORKER_HOSTS="10.20.1.103 10.20.1.101" \
|
|
DASK_ADDRESS="tcp://10.20.1.103:8786" \
|
|
NWORKERS=84 \
|
|
NTHREADS=1 \
|
|
MPIEXEC_FULL="mpirun -np 4 -hostfile /home/yx/qibotn/hostfile -perhost 2" \
|
|
tools/run_tn_dask_mpi_all.sh
|
|
|
|
|
|
|
|
I_MPI_FABRICS=shm:ofi \
|
|
I_MPI_OFI_PROVIDER=tcp \
|
|
FI_PROVIDER=tcp \
|
|
CASE=main1 \
|
|
OBSERVABLES=long_z_string \
|
|
NQUBITS=34 \
|
|
NLAYERS=20 \
|
|
TORCH_THREADS=48 \
|
|
SEARCH_REPEATS=2048 \
|
|
SEARCH_TIME=300 \
|
|
SCHEDULER_HOST=127.0.0.1 \
|
|
WORKER_HOSTS="127.0.0.1" \
|
|
DASK_ADDRESS="tcp://127.0.0.1:8786" \
|
|
NWORKERS=12 \
|
|
NTHREADS=1 \
|
|
MPIEXEC_FULL="mpirun -np 2 -hostfile /home/yx/qibotn/hostfile -perhost 2" \
|
|
tools/run_tn_dask_mpi_all.sh
|
|
# 单独缩并contract计算
|
|
|
|
I_MPI_FABRICS=shm:ofi \
|
|
I_MPI_OFI_PROVIDER=tcp \
|
|
FI_PROVIDER=tcp \
|
|
mpirun -np 4 -hostfile /home/yx/qibotn/hostfile -perhost 2 \
|
|
.venv/bin/python -u tools/tn_contest_runner.py contract \
|
|
--mpi \
|
|
--case main1 \
|
|
--nqubits 34 \
|
|
--nlayers 20 \
|
|
--observables long_z_string \
|
|
--tree-dir trees/contest_tn \
|
|
--torch-threads 48 \
|
|
--dtype complex64
|
|
```
|
|
|
|
# MPS
|
|
```
|
|
cd /home/yx/qibotn
|
|
|
|
I_MPI_FABRICS=shm:ofi \
|
|
I_MPI_OFI_PROVIDER=tcp \
|
|
FI_PROVIDER=tcp \
|
|
MPIEXEC_FULL="mpirun -np 2 -hostfile /home/yx/qibotn/hostfile -perhost 2" \
|
|
TORCH_THREADS=48 \
|
|
OBS_FILTER=ring_xz \
|
|
MAIN1_NQ=128 \
|
|
MAIN1_LAYERS=24 \
|
|
MAIN1_BOND=512 \
|
|
tools/run_vidal_mpi_contest_cases.sh main1
|
|
```
|
|
|
|
|
|
136.997
|
|
|
|
|