#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 |