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

Jump to bottom

Expose noise amplitude for SCIF #383

Merged
merged 6 commits into from
Sep 27, 2022
Merged

Expose noise amplitude for SCIF #383

Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
60a2d94
Enable exception proc_map, working with dataclasses, etc.
GaboFGuerra Sep 26, 2022
91203d8
Integrate YS's changes to enable req_pause.
GaboFGuerra Sep 26, 2022
6574bba
Expose noise amplitude at the process level for SCIF.
GaboFGuerra Sep 27, 2022
536d5f4
Update with latest changes from upstream repo.
GaboFGuerra Sep 27, 2022
1c6e3bb
Remove prints from req_pause debug.
GaboFGuerra Sep 27, 2022
4288c5a
Solve lint issue.
GaboFGuerra Sep 27, 2022
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
16 changes: 10 additions & 6 deletions src/lava/proc/scif/process.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,8 @@ def __init__(
shape: ty.Tuple[int, ...],
step_size: ty.Optional[int] = 1,
theta: ty.Optional[int] = 4,
neg_tau_ref: ty.Optional[int] = -5) -> None:
neg_tau_ref: ty.Optional[int] = -5,
srrisbud marked this conversation as resolved.
Show resolved Hide resolved
noise_amplitude: ty.Optional[int] = 0) -> None:
"""
Stochastic Constraint Integrate and Fire neuron Process.
Expand Down Expand Up @@ -51,8 +52,7 @@ def __init__(
int))
self.state = Var(shape=shape, init=np.zeros(shape=shape).astype(int))
self.spk_hist = Var(shape=shape, init=np.zeros(shape=shape).astype(int))
self.noise_ampl = Var(shape=shape, init=np.zeros(
shape=shape).astype(int))
self.noise_ampl = Var(shape=shape, init=noise_amplitude)

self.step_size = Var(shape=shape, init=int(step_size))
self.theta = Var(shape=(1,), init=int(theta))
Expand All @@ -72,12 +72,14 @@ def __init__(self,
shape: ty.Tuple[int, ...],
step_size: ty.Optional[int] = 1,
theta: ty.Optional[int] = 4,
neg_tau_ref: ty.Optional[int] = -5):
neg_tau_ref: ty.Optional[int] = -5,
noise_amplitude: ty.Optional[int] = 0):

super(CspScif, self).__init__(shape=shape,
step_size=step_size,
theta=theta,
neg_tau_ref=neg_tau_ref)
neg_tau_ref=neg_tau_ref,
noise_amplitude=noise_amplitude)


class QuboScif(AbstractScif):
Expand All @@ -91,10 +93,12 @@ def __init__(self,
cost_diag: npty.NDArray,
step_size: ty.Optional[int] = 1,
theta: ty.Optional[int] = 4,
neg_tau_ref: ty.Optional[int] = -5):
neg_tau_ref: ty.Optional[int] = -5,
noise_amplitude: ty.Optional[int] = 0):

super(QuboScif, self).__init__(shape=shape,
step_size=step_size,
theta=theta,
noise_amplitude=noise_amplitude,
neg_tau_ref=neg_tau_ref)
self.cost_diagonal = Var(shape=shape, init=cost_diag)