blob: 48a0c5b193bb167017cb900ca5eb6abc324e37b7 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
|
# public api
## benchmark.c
The public API for running crossbar simulation, calculating performance metrics,
and generating plots.
* **MemModel struct**
* This is a struct for making model swapping easier
* members
* const char *model_path
* const char *subcircuit_name
* **run_benchmark() (int)**
* Executes the full crossbar simulation pipeline and stores the decoded
crossbar predictions in the passed output array.
* parameters
* const spires_reservoir_config *config
* spires_reservoir *reservoir
* Reservoir_State_Matrix *state_matrix
* const char *model_path
* const char *subcircuit_name
* double *predictions_out
* **calculate_MSE() (double)**
* calculates the mean squared error of the predictions against the expected
values
* parameters
* const double *expected
* const double *predicted
* const size_t *num_steps
* const size_t num_outputs
* **plot_raster() (int)**
* generates a raster plot of the reservoir based on the spike threshhold parameter
(not that useful tbh but can be cool to see), plot stored as svg in output directory
* parameters
* const Reservoir_State_Matrix *matrix
* const size_t neurons_to_plot
* const double spike_threshold
* **plot_reservoir_predictions() (int)**
* plots the decoded crossbar predictions against the expected values, plot stored
as svg in output directory with name "reservoir_prediction_$(model_name).svg"
* parameters
* const double *expected
* const double *predicted
* size_t num_samples
* size_t num_outputs
* size_t output_to_plot
* const char *model_path
* **plot_model_delta() (int)**
* plots the prediction differences for every timestep between the given model
and the fixed resistor baseline
* parameters
* const double *fixed
* const double *model
* const size_t num_samples
* const size_t num_outputs
* const size_t output_to_plot
# Internal header files
## crossbar_generator.h
Handles generation of the SPICE netlist used to simulate the memristive crossbar.
* **Crossbar_Config struct**
* Stores the configuration needed to generate a crossbar netlist
* members
* size_t rows
* size_t columns
* const double *input_series
* size_t num_samples
* const double *initial_resistance
* double load_resistance
* const char *model_path
* const char *subcircuit_name
* double time_step
* double stop_time
* int print_state_nodes
* **generate_crossbar() (int)**
* Generates the SPICE crossbar netlist using the passed Crossbar_Config
* parameters
* const char *output_filename
* const Crossbar_Config *config
## spires_interface.h
Handles communication between SPIRES and the crossbar simulation, including
collecting reservoir states and mapping trained weights to resistances.
* **Reservoir_State_Matrix struct**
* Stores the collected reservoir states in a flat row-major array
* members
* size_t num_samples
* size_t num_features
* double *states
* **conductance_mapping struct**
* Stores the values used to map software readout weights to crossbar
conductances and later decode the crossbar outputs
* members
* double g_min
* double g_max
* double alpha
* double max_abs_weight
* **collect_reservoir_states() (int)**
* Runs the input series through the SPIRES reservoir and saves the state of
every neuron at every timestep
* parameters
* spires_reservoir *reservoir
* const double *input_series
* size_t series_length
* Reservoir_State_Matrix *result
* **convert_weights_to_resistances() (int)**
* Converts the trained SPIRES readout weights into differential-pair
resistances for the crossbar
* parameters
* const spires_reservoir *reservoir
* size_t num_neurons
* size_t num_outputs
* double r_on
* double r_off
* double **resistances_out
* conductance_mapping *mapping
* **train_reservoir() (int)**
* Wrapper around SPIRES ridge regression training
* parameters
* spires_reservoir *reservoir
* double *input_series
* double *target_series
* size_t series_length
* double lambda
* **free_reservoir_state_matrix() (void)**
* Frees the memory allocated for a Reservoir_State_Matrix
* parameters
* Reservoir_State_Matrix *matrix
## read_crossbar.h
Handles running ngspice, reading the generated simulation data, and decoding the
crossbar voltages back into software predictions.
* **Crossbar_Output_Matrix struct**
* Stores the output voltages read from the ngspice simulation
* members
* size_t num_samples
* size_t num_outputs
* double *time
* double *voltages
* **run_ngspice() (int)**
* Runs ngspice in batch mode on the generated crossbar netlist
* parameters
* const char *crossbar_path
* **read_crossbar() (int)**
* Reads the ngspice output data file into a Crossbar_Output_Matrix
* parameters
* const char *data_path
* size_t num_outputs
* Crossbar_Output_Matrix *result
* **convert_output_to_software() (int)**
* Decodes the differential-pair crossbar voltages back into the equivalent
software readout predictions
* parameters
* size_t num_neurons
* size_t num_outputs
* size_t num_timesteps
* const double *voltages
* const double *resistances
* double load_resistance
* const conductance_mapping *mapping
* const double *row_voltages
* double spike_amplitude
* double *decoded_outputs
* **free_crossbar_output_matrix() (void)**
* Frees the memory allocated for a Crossbar_Output_Matrix
* parameters
* Crossbar_Output_Matrix *result
|
