Skip to content

Commit

Permalink
fix: removed toplevel cuquantum import
Browse files Browse the repository at this point in the history
  • Loading branch information
BrunoLiegiBastonLiegi committed Apr 10, 2024
1 parent ed74bef commit 72f4841
Showing 1 changed file with 13 additions and 11 deletions.
24 changes: 13 additions & 11 deletions src/qibotn/backends/cutensornet.py
Original file line number Diff line number Diff line change
@@ -1,19 +1,9 @@
import cuquantum # pylint: disable=import-error
import numpy as np
from qibo.backends.numpy import NumpyBackend
from qibo.config import raise_error
from qibo.result import QuantumState

CUDA_TYPES = {
"complex64": (
cuquantum.cudaDataType.CUDA_C_32F,
cuquantum.ComputeType.COMPUTE_32F,
),
"complex128": (
cuquantum.cudaDataType.CUDA_C_64F,
cuquantum.ComputeType.COMPUTE_64F,
),
}
CUDA_TYPES = {}


class CuTensorNet(NumpyBackend): # pragma: no cover
Expand Down Expand Up @@ -75,6 +65,18 @@ def __init__(self, runcard):
self.supports_multigpu = True
self.handle = self.cutn.create()

global CUDA_TYPES
CUDA_TYPES = {
"complex64": (
self.cuquantum.cudaDataType.CUDA_C_32F,
self.cuquantum.ComputeType.COMPUTE_32F,
),
"complex128": (
self.cuquantum.cudaDataType.CUDA_C_64F,
self.cuquantum.ComputeType.COMPUTE_64F,
),
}

def apply_gate(self, gate, state, nqubits): # pragma: no cover
raise_error(NotImplementedError, "QiboTN cannot apply gates directly.")

Expand Down

0 comments on commit 72f4841

Please sign in to comment.