From a4edd40c35e2784691936662a8a89463cc434f9a Mon Sep 17 00:00:00 2001 From: Your Name Date: Tue, 21 Jul 2026 10:20:48 -0700 Subject: Spires feeding into SPICE crossbar --- src/spires_interface_test.c | 218 ++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 218 insertions(+) create mode 100644 src/spires_interface_test.c (limited to 'src/spires_interface_test.c') diff --git a/src/spires_interface_test.c b/src/spires_interface_test.c new file mode 100644 index 0000000..ef2dabc --- /dev/null +++ b/src/spires_interface_test.c @@ -0,0 +1,218 @@ +#include "spires_interface.h" +#include "crossbar_generator.h" + +#include +#include +#include +#include +#include + +#define NUM_NEURONS 400 +#define NUM_INPUTS 4 +#define NUM_OUTPUTS 2 +#define NUM_TRAINING_STEPS 500 + +#define PI 3.14159265358979323846 + +static int plot_raster( + const Reservoir_State_Matrix *matrix, + size_t neurons_to_plot, + double spike_threshold +); + +int main(void) { + //discrete LIF parameters + double lif_config[] = { + 0.0, //V_off + 1.0, //V_th + 0.2, //leak rate + 0.5, //bias + }; + + const spires_reservoir_config config = { + .num_neurons = NUM_NEURONS, + .num_inputs = NUM_INPUTS, + .num_outputs = NUM_OUTPUTS, + .spectral_radius = 0.95, + .ei_ratio = 0.8, + .input_strength = 0.1, + .connectivity = 0.1, + .dt = 1.0, + .connectivity_type = SPIRES_CONN_RANDOM, + .neuron_type = SPIRES_NEURON_LIF_DISCRETE, + .neuron_params = lif_config + }; + + spires_reservoir *reservoir = NULL; + + spires_status status = spires_reservoir_create( + &config, + &reservoir + ); + + if (status != SPIRES_OK) { + fprintf(stderr, "Failed to create reservoir"); + return -1; + } + + //create data set (sin wave time series prediction) + double training_inputs[NUM_TRAINING_STEPS * NUM_INPUTS]; + for (size_t timestep = 0; timestep < NUM_TRAINING_STEPS; timestep++) { + for (size_t input = 0; input < NUM_INPUTS; input++) { + training_inputs[timestep * NUM_INPUTS + input] = sin(2.0 * PI * (double)timestep / 50.0); + } + } + + Reservoir_State_Matrix state_matrix = {0}; + if (collect_reservoir_states(reservoir, training_inputs, NUM_TRAINING_STEPS, + &state_matrix) != 0) { + fprintf(stderr, "Failed to collect reservoir states"); + spires_reservoir_destroy(reservoir); + return -1; + } + + printf("collected state matrix: %zu x %zu\n", state_matrix.num_samples, + state_matrix.num_features); + + //generate raster plot for verification + if (plot_raster(&state_matrix, NUM_NEURONS, 0.5) != 0) { + fprintf(stderr, "Failed to plot raster\n"); + } + + //now I need to send the outputs to a crossbar + double *initial_resistances = malloc(NUM_NEURONS * NUM_OUTPUTS * sizeof(*initial_resistances)); + if (!initial_resistances) { + fprintf(stderr, "Failed to allocate memory for initial resistances"); + return -1; + } + + //all memristors start with same resistance for now + for (size_t i = 0; i < NUM_NEURONS * NUM_OUTPUTS; i++) { + initial_resistances[i] = 80000; + } + + const Crossbar_Config crossbar_config = { + .rows = state_matrix.num_features, + .columns = NUM_OUTPUTS, + .input_series = state_matrix.states, + .num_samples = state_matrix.num_samples, + .initial_resistance = initial_resistances, + .model_path = "hp_memristor.cir", + .subcircuit_name = "memristor", + .load_resistance = 50.0, + .time_step = 1e-6, + .stop_time = state_matrix.num_samples * 1e-6, + .print_state_nodes = 0 + }; + + if (generate_crossbar("crossbar.cir", &crossbar_config) < 0) { + fprintf(stderr, "failed to create crossbar config"); + } + + //call ngspice for crossbar + + //clean up + printf("YAY IT WORKED!!! Cleaning up :)"); + free(initial_resistances); + free_reservoir_state_matrix(&state_matrix); + spires_reservoir_destroy(reservoir); + + return 0; +} + +static int plot_raster( + const Reservoir_State_Matrix *matrix, + size_t neurons_to_plot, + double spike_threshold +) { + if (!matrix || !matrix->states || matrix->num_samples == 0) { + return -1; + } + + if (neurons_to_plot > matrix->num_features) { + neurons_to_plot = matrix->num_features; + } + + /* first pass: count spikes */ + size_t spike_count = 0; + for (size_t t = 0; t < matrix->num_samples; t++) { + for (size_t n = 0; n < neurons_to_plot; n++) { + double value = matrix->states[t * matrix->num_features + n]; + if (value > spike_threshold) { + spike_count++; + } + } + } + + if (spike_count == 0) { + fprintf(stderr, "No spikes found above threshold %.3f\n", spike_threshold); + return -1; + } + + PLFLT *x = malloc(spike_count * sizeof(*x)); + PLFLT *y = malloc(spike_count * sizeof(*y)); + if (!x || !y) { + free(x); + free(y); + return -1; + } + + /* second pass: fill spike coordinates */ + size_t k = 0; + for (size_t t = 0; t < matrix->num_samples; t++) { + for (size_t n = 0; n < neurons_to_plot; n++) { + double value = matrix->states[t * matrix->num_features + n]; + if (value > spike_threshold) { + x[k] = (PLFLT)t; + y[k] = (PLFLT)n; + k++; + } + } + } + + /* pick whichever device you have */ + plsdev("pngcairo"); + plsfnam("reservoir_raster.png"); + + plsetopt("geometry", "1600x1200"); + plscolbg(255, 255, 255); + + plinit(); + + plscol0(1, 40, 40, 40); //gray axis + plscol0(2, 0, 0, 0); //blue points + + plcol0(1); + plwidth(1.0); + + plenv( + 0.0, + (PLFLT)(matrix->num_samples - 1), + 0.0, + (PLFLT)(neurons_to_plot - 1), + 0, + 0 + ); + + pllab( + "Timestep", + "Neuron index", + "SPIRES Reservoir Raster Plot" + ); + + plcol0(2); + plwidth(1.0); + + for (size_t i = 0; i < spike_count; i++) { + PLFLT xline[2] = {x[i], x[i]}; + PLFLT yline[2] = {y[i] - 0.35, y[i] + 0.35}; + + plline(2, xline, yline); + } + + plend(); + + free(x); + free(y); + return 0; +} -- cgit v1.2.3