#include <vector>
#include <cuda.h>
#include <cuda_runtime.h>
#include "interface.h"
Go to the source code of this file.
|
void | findBestMatchOnDevice (dim3 grid, dim3 block, float *outputs, int *winner, int numOutputs) |
| Wrapper for findBestMatchPass1Kernel and findBestMatchPass2Kernel.
|
|
void | propogateInputOnDevice (dim3 grid, dim3 block, float *inputs, float *weights, float *outputs, int numInputs, int sequenceId, int numOutputs) |
| Wrapper for propogateInputKernel.
|
|
void | updateWeightsOnDevice (dim3 grid, dim3 block, float *inputs, float *weights, int *winner, float sigma, int numInputs, int sequenceId, int numOutputs, int neighbourhoodSize, float initLearningRate, int numIterations, int currentIteration) |
| Wrapper for updateWeightsKernel.
|
|
#define MAX_GPU_DEVICES 4 |
void findBestMatchOnDevice |
( |
dim3 |
grid, |
|
|
dim3 |
block, |
|
|
float * |
outputs, |
|
|
int * |
winner, |
|
|
int |
numOutputs |
|
) |
| |
Wrapper for findBestMatchPass1Kernel and findBestMatchPass2Kernel.
- Parameters
-
[in] | grid | - CUDA grid size |
[in] | block | - CUDA block size |
[in] | outputs | - outputs |
[in] | numOutputs | - number of outputs |
[out] | winner | - winner |
void propogateInputOnDevice |
( |
dim3 |
grid, |
|
|
dim3 |
block, |
|
|
float * |
inputs, |
|
|
float * |
weights, |
|
|
float * |
outputs, |
|
|
int |
numInputs, |
|
|
int |
sequenceId, |
|
|
int |
numOutputs |
|
) |
| |
Wrapper for propogateInputKernel.
- Parameters
-
[in] | grid | - CUDA grid size |
[in] | block | - CUDA block size |
[in] | inputs | - inputs |
[in] | weights | - weights |
[in] | numInputs | - number of inputs |
[in] | sequenceId | - sequence id |
[in] | numOutputs | - number of outputs |
[out] | outputs | - outputs |
void updateWeightsOnDevice |
( |
dim3 |
grid, |
|
|
dim3 |
block, |
|
|
float * |
inputs, |
|
|
float * |
weights, |
|
|
int * |
winner, |
|
|
float |
sigma, |
|
|
int |
numInputs, |
|
|
int |
sequenceId, |
|
|
int |
numOutputs, |
|
|
int |
neighbourhoodSize, |
|
|
float |
initLearningRate, |
|
|
int |
numIterations, |
|
|
int |
currentIteration |
|
) |
| |
Wrapper for updateWeightsKernel.
- Parameters
-
[in] | grid | - CUDA grid size |
[in] | block | - CUDA block size |
[in] | inputs | - inputs |
[in] | winner | - winner |
[in] | sigma | - sigma |
[in] | numInputs | - number of inputs |
[in] | sequenceId | - sequence id |
[in] | numOutputs | - number of outputs |
[in] | neighbourhoodSize | - neighbourhood size |
[in] | numIterations | - number of iterations |
[in] | currentIteration | - current iteration |
[out] | weights | - weights |