mtjrider

import typing
import numpy as np
import cupy as cp
import torch

from cuquantum import custatevec as cusv
from cuquantum import cudaDataType, ComputeType


nSVs          = 1

mtjrider

import torch
import numpy as np
import cuquantum
from cuquantum import custatevec as cusv


num_qubits = 24
device = "cuda:0"
dtype = torch.complex64
cusv_state_dtype = cuquantum.cudaDataType.CUDA_C_32F

mtjrider

# This isn't supposed to run as a bash script, i named it with ".sh" for syntax highlighting.

# https://developer.nvidia.com/nsight-systems
# https://docs.nvidia.com/nsight-systems/profiling/index.html

# My preferred nsys (command line executable used to create profiles) commands
#
# In your script, write
# torch.cuda.nvtx.range_push("region name")
# ...