diff options
Diffstat (limited to 'src/spires_interface_test.c')
| -rw-r--r-- | src/spires_interface_test.c | 218 |
1 files changed, 0 insertions, 218 deletions
diff --git a/src/spires_interface_test.c b/src/spires_interface_test.c deleted file mode 100644 index ef2dabc..0000000 --- a/src/spires_interface_test.c +++ /dev/null @@ -1,218 +0,0 @@ -#include "spires_interface.h" -#include "crossbar_generator.h" - -#include <math.h> -#include <spires.h> -#include <stdio.h> -#include <stdlib.h> -#include <plplot/plplot.h> - -#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; -} |