~speedprog/mtg/mtg_card_detector.git

summary
reflog
commits
tree
tickets
docs
forks
compare

Joseph Redmon

2015-11-26 0305fb4d99cf1efc7d4aa4d2ee2d65d54500d437

 src/softmax_layer_kernels.cu

@@ -1,11 +1,13 @@
#include "cuda_runtime.h"
#include "curand.h"
#include "cublas_v2.h"

extern "C" {
#include "softmax_layer.h"
#include "cuda.h"
#include "blas.h"
}

#define BLOCK 256

__global__ void forward_softmax_layer_kernel(int n, int batch, float *input, float *output)
{
    int b = (blockIdx.x + blockIdx.y*gridDim.x) * blockDim.x + threadIdx.x;

			@@ -1,11 +1,13 @@
			#include "cuda_runtime.h"
			#include "curand.h"
			#include "cublas_v2.h"

			extern "C" {
			#include "softmax_layer.h"
			#include "cuda.h"
			#include "blas.h"
			}

			#define BLOCK 256

			__global__ void forward_softmax_layer_kernel(int n, int batch, float input, float output)
			{
			int b = (blockIdx.x + blockIdx.ygridDim.x) blockDim.x + threadIdx.x;