| | |
|---|
| | | #include "cuda_runtime.h" |
|---|
| | | #include "curand.h" |
|---|
| | | #include "cublas_v2.h" |
|---|
| | | |
|---|
| | | extern "C" { |
|---|
| | | #include "dropout_layer.h" |
|---|
| | | #include "cuda.h" |
|---|
| | | #include "utils.h" |
|---|
| | | #include "params.h" |
|---|
| | | } |
|---|
| | | |
|---|
| | | __global__ void yoloswag420blazeit360noscope(float *input, int size, float *rand, float prob, float scale) |
|---|
| | |
|---|
| | | if (!state.train) return; |
|---|
| | | int size = layer.inputs*layer.batch; |
|---|
| | | cuda_random(layer.rand_gpu, size); |
|---|
| | | /* |
|---|
| | | int i; |
|---|
| | | for(i = 0; i < size; ++i){ |
|---|
| | | layer.rand[i] = rand_uniform(); |
|---|
| | | } |
|---|
| | | cuda_push_array(layer.rand_gpu, layer.rand, size); |
|---|
| | | */ |
|---|
| | | |
|---|
| | | yoloswag420blazeit360noscope<<<cuda_gridsize(size), BLOCK>>>(state.input, size, layer.rand_gpu, layer.probability, layer.scale); |
|---|
| | | check_error(cudaPeekAtLastError()); |
|---|
