Skip to content

Commit

Permalink
fix: review updates
Browse files Browse the repository at this point in the history
  • Loading branch information
BrunoLiegiBastonLiegi committed Feb 7, 2024
1 parent 4e6e481 commit 1c49fc4
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 6 deletions.
4 changes: 2 additions & 2 deletions src/qibojit/backends/cpu.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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):
Expand Down
8 changes: 4 additions & 4 deletions src/qibojit/backends/gpu.py
Original file line number Diff line number Diff line change
@@ -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

Expand Down Expand Up @@ -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):
Expand Down Expand Up @@ -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)

Expand Down

0 comments on commit 1c49fc4

Please sign in to comment.