From 1c49fc4ac49f9216e0a5762d6105543db7b659d2 Mon Sep 17 00:00:00 2001 From: Andrea Papaluca Date: Wed, 7 Feb 2024 12:23:14 +0400 Subject: [PATCH] fix: review updates --- src/qibojit/backends/cpu.py | 4 ++-- src/qibojit/backends/gpu.py | 8 ++++---- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/src/qibojit/backends/cpu.py b/src/qibojit/backends/cpu.py index ed0276a9..315c44f5 100644 --- a/src/qibojit/backends/cpu.py +++ b/src/qibojit/backends/cpu.py @@ -2,7 +2,7 @@ import numpy as np from numba import njit -from qibo.backends import clifford_operations +from qibo.backends import _clifford_operations from qibo.backends.numpy import NumpyBackend from qibo.config import log from qibo.gates.abstract import ParametrizedGate @@ -74,7 +74,7 @@ def __init__(self): else: self.set_threads(len(psutil.Process().cpu_affinity())) - spec = find_spec("qibo.backends.clifford_operations") + spec = find_spec("qibo.backends._clifford_operations") self.clifford_operations = module_from_spec(spec) spec.loader.exec_module(self.clifford_operations) for method in dir(clifford_operations_cpu): diff --git a/src/qibojit/backends/gpu.py b/src/qibojit/backends/gpu.py index 5d274a2e..f4f041fb 100644 --- a/src/qibojit/backends/gpu.py +++ b/src/qibojit/backends/gpu.py @@ -1,7 +1,7 @@ from importlib.util import find_spec, module_from_spec import numpy as np -from qibo.backends import clifford_operations +from qibo.backends import _clifford_operations from qibo.backends.numpy import NumpyBackend from qibo.config import log, raise_error @@ -103,7 +103,7 @@ def kernel_loader(name, ktype): from qibojit.backends import clifford_operations_gpu - spec = find_spec("qibo.backends.clifford_operations") + spec = find_spec("qibo.backends._clifford_operations") self.clifford_operations = module_from_spec(spec) spec.loader.exec_module(self.clifford_operations) for method in dir(clifford_operations_gpu): @@ -158,10 +158,10 @@ def to_numpy(self, x): def issparse(self, x): return self.sparse.issparse(x) or self.npsparse.issparse(x) - def clifford_pre_execution_reshape(self, state): + def _clifford_pre_execution_reshape(self, state): return state.ravel() - def clifford_post_execution_reshape(self, state, nqubits): + def _clifford_post_execution_reshape(self, state, nqubits): dim = 2 * nqubits + 1 return state.reshape(dim, dim)