Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add stim.FlipSimulator.broadcast_pauli_errors #678

Merged
merged 7 commits into from
Dec 11, 2023
Merged

Add stim.FlipSimulator.broadcast_pauli_errors #678

Show file tree
Hide file tree
Changes from 3 commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
f556295
add apply_pauli_errors
m-mcewen Dec 7, 2023
62c0c1a
bittwidling, also doctests
m-mcewen Dec 7, 2023
73c3d79
fix pybind tests
m-mcewen Dec 7, 2023
4044b76
craig review
m-mcewen Dec 8, 2023
2feae5a
prevent running with scisors
m-mcewen Dec 8, 2023
582ade0
...regen_docs
m-mcewen Dec 8, 2023
03c4306
Finish up
Strilanc Dec 10, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
39 changes: 39 additions & 0 deletions doc/python_api_reference_vDev.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -172,6 +172,7 @@ API references for stable versions are kept on the [stim github wiki](https://gi
- [`stim.ExplainedError.dem_error_terms`](#stim.ExplainedError.dem_error_terms)
- [`stim.FlipSimulator`](#stim.FlipSimulator)
- [`stim.FlipSimulator.__init__`](#stim.FlipSimulator.__init__)
- [`stim.FlipSimulator.apply_pauli_errors`](#stim.FlipSimulator.apply_pauli_errors)
- [`stim.FlipSimulator.batch_size`](#stim.FlipSimulator.batch_size)
- [`stim.FlipSimulator.do`](#stim.FlipSimulator.do)
- [`stim.FlipSimulator.get_detector_flips`](#stim.FlipSimulator.get_detector_flips)
Expand Down Expand Up @@ -6131,6 +6132,44 @@ def __init__(
"""
```

<a name="stim.FlipSimulator.apply_pauli_errors"></a>
```python
# stim.FlipSimulator.apply_pauli_errors

# (in class stim.FlipSimulator)
def apply_pauli_errors(
This conversation was marked as resolved.
Show resolved Hide resolved
self,
*,
pauli: Union[str, int],
mask: np.ndarray,
) -> None:
"""Applies a pauli over all qubits in all simulation indices, filtered by mask.
This conversation was marked as resolved.
Show resolved Hide resolved

Args:
pauli: The pauli, specified as an integer or string.
Uses the convention 0=I, 1=X, 2=Y, 3=Z.
Any value from [0, 1, 2, 3, 'X', 'Y', 'Z', 'I', '_'] is allowed.
mask: a bool array with shape (qubit, simulation_instance)
This conversation was marked as resolved.
Show resolved Hide resolved
The pauli error is only applied to qubits q and simulation indices k
where mask[q, k] == True

Examples:
>>> import stim
>>> import numpy as np
>>> sim = stim.FlipSimulator(
... batch_size=2,
... num_qubits=3,
... disable_stabilizer_randomization=True,
... )
>>> sim.apply_pauli_errors(
This conversation was marked as resolved.
Show resolved Hide resolved
... pauli='X',
... mask=np.asarray([[True, False],[False, False],[True, True]]),
... )
>>> sim.peek_pauli_flips()
[stim.PauliString("+X_X"), stim.PauliString("+__X")]
"""
```

<a name="stim.FlipSimulator.batch_size"></a>
```python
# stim.FlipSimulator.batch_size
Expand Down
31 changes: 31 additions & 0 deletions doc/stim.pyi
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4655,6 +4655,37 @@ class FlipSimulator:
>>> import stim
>>> sim = stim.FlipSimulator(batch_size=256)
"""
def apply_pauli_errors(
self,
*,
pauli: Union[str, int],
mask: np.ndarray,
) -> None:
"""Applies a pauli over all qubits in all simulation indices, filtered by mask.

Args:
pauli: The pauli, specified as an integer or string.
Uses the convention 0=I, 1=X, 2=Y, 3=Z.
Any value from [0, 1, 2, 3, 'X', 'Y', 'Z', 'I', '_'] is allowed.
mask: a bool array with shape (qubit, simulation_instance)
The pauli error is only applied to qubits q and simulation indices k
where mask[q, k] == True

Examples:
>>> import stim
>>> import numpy as np
>>> sim = stim.FlipSimulator(
... batch_size=2,
... num_qubits=3,
... disable_stabilizer_randomization=True,
... )
>>> sim.apply_pauli_errors(
... pauli='X',
... mask=np.asarray([[True, False],[False, False],[True, True]]),
... )
>>> sim.peek_pauli_flips()
[stim.PauliString("+X_X"), stim.PauliString("+__X")]
"""
@property
def batch_size(
self,
Expand Down
31 changes: 31 additions & 0 deletions glue/python/src/stim/__init__.pyi
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4655,6 +4655,37 @@ class FlipSimulator:
>>> import stim
>>> sim = stim.FlipSimulator(batch_size=256)
"""
def apply_pauli_errors(
self,
*,
pauli: Union[str, int],
mask: np.ndarray,
) -> None:
"""Applies a pauli over all qubits in all simulation indices, filtered by mask.

Args:
pauli: The pauli, specified as an integer or string.
Uses the convention 0=I, 1=X, 2=Y, 3=Z.
Any value from [0, 1, 2, 3, 'X', 'Y', 'Z', 'I', '_'] is allowed.
mask: a bool array with shape (qubit, simulation_instance)
The pauli error is only applied to qubits q and simulation indices k
where mask[q, k] == True

Examples:
>>> import stim
>>> import numpy as np
>>> sim = stim.FlipSimulator(
... batch_size=2,
... num_qubits=3,
... disable_stabilizer_randomization=True,
... )
>>> sim.apply_pauli_errors(
... pauli='X',
... mask=np.asarray([[True, False],[False, False],[True, True]]),
... )
>>> sim.peek_pauli_flips()
[stim.PauliString("+X_X"), stim.PauliString("+__X")]
"""
@property
def batch_size(
self,
Expand Down
87 changes: 87 additions & 0 deletions src/stim/simulators/frame_simulator.pybind.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -770,4 +770,91 @@ void stim_pybind::pybind_frame_simulator_methods(
[stim.PauliString("+YX__")]
)DOC")
.data());

c.def(
"apply_pauli_errors",
[](FrameSimulator<MAX_BITWORD_WIDTH> &self,
const pybind11::object &pauli,
const pybind11::object &mask
) {
uint8_t p = 255;
try {
p = pybind11::cast<uint8_t>(pauli);
} catch (const pybind11::cast_error &) {
try {
std::string s = pybind11::cast<std::string>(pauli);
if (s == "X") {
p = 1;
} else if (s == "Y") {
p = 2;
} else if (s == "Z") {
p = 3;
} else if (s == "I" || s == "_") {
p = 0;
}
} catch (const pybind11::cast_error &) {
}
}

bool flip_z_part = p & 2;
bool flip_x_part = 6 >> p & 1; // 0b0110 >> p & 0b0001
This conversation was marked as resolved.
Show resolved Hide resolved

if (pybind11::isinstance<pybind11::array_t<bool>>(mask)) {
// can use pybind11::isinstance<pybind11::array_t<bool, pybind11::array::c_style>>
// to check for dense c_ordered array for copying, if this is too slow for you
const pybind11::array_t<bool> &arr = pybind11::cast<pybind11::array_t<bool>>(mask);
if (arr.ndim() == 2) {
//pybind11::ssize_t???
This conversation was marked as resolved.
Show resolved Hide resolved
size_t major = arr.shape(0);
size_t minor = arr.shape(1);
//size_t major_stride = arr.strides(0);
//size_t minor_stride = arr.strides(1);
auto u = arr.unchecked<2>();
for (size_t i = 0; i < major; i++){
for (size_t j = 0; j < minor; j++){
auto b = u.data(i, j);
self.x_table[i][j] ^= *b & flip_x_part;
self.z_table[i][j] ^= *b & flip_z_part;
}
}
} else {
throw std::invalid_argument(
This conversation was marked as resolved.
Show resolved Hide resolved
"apply_pauli_errors currently only supports a mask that is a 2D bool numpy array");
This conversation was marked as resolved.
Show resolved Hide resolved
}
} else {
throw std::invalid_argument(
Strilanc marked this conversation as resolved.
Show resolved Hide resolved
"apply_pauli_errors currently only supports a mask that is a 2D bool numpy array");
}
},
pybind11::kw_only(),
pybind11::arg("pauli"),
pybind11::arg("mask"),
clean_doc_string(R"DOC(
@signature def apply_pauli_errors(self, *, pauli: Union[str, int], mask: np.ndarray) -> None:
Applies a pauli over all qubits in all simulation indices, filtered by mask.

Args:
pauli: The pauli, specified as an integer or string.
Uses the convention 0=I, 1=X, 2=Y, 3=Z.
Any value from [0, 1, 2, 3, 'X', 'Y', 'Z', 'I', '_'] is allowed.
mask: a bool array with shape (qubit, simulation_instance)
The pauli error is only applied to qubits q and simulation indices k
where mask[q, k] == True

Examples:
>>> import stim
>>> import numpy as np
>>> sim = stim.FlipSimulator(
... batch_size=2,
... num_qubits=3,
... disable_stabilizer_randomization=True,
... )
>>> sim.apply_pauli_errors(
... pauli='X',
... mask=np.asarray([[True, False],[False, False],[True, True]]),
... )
>>> sim.peek_pauli_flips()
[stim.PauliString("+X_X"), stim.PauliString("+__X")]
)DOC")
.data());
}
92 changes: 92 additions & 0 deletions src/stim/simulators/frame_simulator_pybind_test.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -216,6 +216,98 @@ def test_set_pauli_flip():
stim.PauliString('XZ___'),
]

def test_apply_pauli_errors():
sim = stim.FlipSimulator(
This conversation was marked as resolved.
Show resolved Hide resolved
batch_size=2,
num_qubits=3,
disable_stabilizer_randomization=True,
)
sim.apply_pauli_errors(
pauli='X',
mask=np.asarray([
[True, False],
[False, False],
[True, True]]
),
)
peek = sim.peek_pauli_flips()
assert peek == [
stim.PauliString("+X_X"),
stim.PauliString("+__X")
]
sim.apply_pauli_errors(
pauli='Z',
mask=np.asarray([
[True, True],
[True, False],
[False, False]]
),
)
peek = sim.peek_pauli_flips()
assert peek == [
stim.PauliString("+YZX"),
stim.PauliString("+Z_X")
]
sim.apply_pauli_errors(
pauli='Y',
mask=np.asarray([
[True, False],
[False, True],
[False, True]]
),
)
peek = sim.peek_pauli_flips()
assert peek == [
stim.PauliString("+_ZX"),
stim.PauliString("+ZYZ")
]

# do it again with ints
sim = stim.FlipSimulator(
batch_size=2,
num_qubits=3,
disable_stabilizer_randomization=True,
)
sim.apply_pauli_errors(
pauli=1,
mask=np.asarray([
[True, False],
[False, False],
[True, True]]
),
)
peek = sim.peek_pauli_flips()
assert peek == [
stim.PauliString("+X_X"),
stim.PauliString("+__X")
]
sim.apply_pauli_errors(
pauli=3,
mask=np.asarray([
[True, True],
[True, False],
[False, False]]
),
)
peek = sim.peek_pauli_flips()
assert peek == [
stim.PauliString("+YZX"),
stim.PauliString("+Z_X")
]
sim.apply_pauli_errors(
pauli=2,
mask=np.asarray([
[True, False],
[False, True],
[False, True]]
),
)
peek = sim.peek_pauli_flips()
assert peek == [
stim.PauliString("+_ZX"),
stim.PauliString("+ZYZ")
]


def test_repro_heralded_pauli_channel_1_bug():
circuit = stim.Circuit("""
Expand Down
Loading