88 lines
2.6 KiB
C
88 lines
2.6 KiB
C
|
#ifndef SPIKEGENERATION2DMANYIP
|
||
|
|
#define SPIKEGENERATION2DMANYIP
|
||
|
|
|
||
|
|
#include <unistd.h>
|
||
|
|
|
||
|
|
#include <cctype>
|
||
|
|
#include <iostream>
|
||
|
|
#include <vector>
|
||
|
|
|
||
|
|
#define SPIKE_GENERATION_NUMBER_OF_KERNELS 1
|
||
|
|
#define SPIKE_GENERATION_NUMBER_OF_KERNELS_PARAMETERS 7
|
||
|
|
|
||
|
|
class SpikeGeneration2DManyIP
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
SpikeGeneration2DManyIP();
|
||
|
|
~SpikeGeneration2DManyIP();
|
||
|
|
|
||
|
|
bool spike_generation_entrypoint(
|
||
|
|
int64_t input_pointer_addr, int64_t input_dim_0,
|
||
|
|
int64_t input_dim_1, int64_t input_dim_2, int64_t input_dim_3,
|
||
|
|
int64_t random_values_pointer_addr, int64_t random_values_dim_0,
|
||
|
|
int64_t random_values_dim_1, int64_t random_values_dim_2,
|
||
|
|
int64_t random_values_dim_3, int64_t output_pointer_addr,
|
||
|
|
int64_t output_dim_0, int64_t output_dim_1, int64_t output_dim_2,
|
||
|
|
int64_t output_dim_3, int64_t number_of_cpu_processes);
|
||
|
|
|
||
|
|
bool spike_generation(
|
||
|
|
float* input_pointer,
|
||
|
|
size_t input_dim_c0,
|
||
|
|
size_t input_dim_c1,
|
||
|
|
size_t input_dim_c2,
|
||
|
|
float* random_values_pointer,
|
||
|
|
size_t random_values_dim_c0,
|
||
|
|
size_t random_values_dim_c1,
|
||
|
|
size_t random_values_dim_c2,
|
||
|
|
int64_t* output_pointer,
|
||
|
|
size_t output_dim_c0,
|
||
|
|
size_t output_dim_c1,
|
||
|
|
size_t output_dim_c2,
|
||
|
|
size_t x_dim,
|
||
|
|
size_t y_dim,
|
||
|
|
size_t spike_dim,
|
||
|
|
size_t h_dim,
|
||
|
|
size_t pattern_id);
|
||
|
|
|
||
|
|
bool gpu_spike_generation(
|
||
|
|
float* input_pointer,
|
||
|
|
size_t input_dim_c0,
|
||
|
|
size_t input_dim_c1,
|
||
|
|
size_t input_dim_c2,
|
||
|
|
float* random_values_pointer,
|
||
|
|
size_t random_values_dim_c0,
|
||
|
|
size_t random_values_dim_c1,
|
||
|
|
size_t random_values_dim_c2,
|
||
|
|
int64_t* output_pointer,
|
||
|
|
size_t output_dim_c0,
|
||
|
|
size_t output_dim_c1,
|
||
|
|
size_t output_dim_c2,
|
||
|
|
size_t x_dim,
|
||
|
|
size_t y_dim,
|
||
|
|
size_t spike_dim,
|
||
|
|
size_t h_dim,
|
||
|
|
size_t number_of_pattern);
|
||
|
|
|
||
|
|
void gpu_occupancy_export(size_t dim_x, size_t dim_y, size_t number_of_pattern,
|
||
|
|
size_t spike_dim, int64_t setting_memory_addr, size_t setting_dim_0, size_t setting_dim_1);
|
||
|
|
|
||
|
|
void gpu_occupancy_import(
|
||
|
|
int64_t setting_memory_addr,
|
||
|
|
size_t setting_dim_0,
|
||
|
|
size_t setting_dim_1);
|
||
|
|
|
||
|
|
private:
|
||
|
|
size_t lower_bound(float* data_ptr, size_t data_length,
|
||
|
|
size_t data_ptr_stride,
|
||
|
|
float compare_to_value);
|
||
|
|
|
||
|
|
void gpu_occupancy_measure(size_t dim_x, size_t dim_y, size_t number_of_pattern,
|
||
|
|
size_t spike_dim);
|
||
|
|
|
||
|
|
bool grid_and_thread_calculated = false;
|
||
|
|
std::vector<std::vector<size_t>> grid_and_thread_settings;
|
||
|
|
bool display_debug = false;
|
||
|
|
};
|
||
|
|
|
||
|
|
#endif /* SPIKEGENERATION2DMANYIP */
|