diff options
| author | sniklaus <simon.niklaus@outlook.com> | 2017-09-09 22:59:59 -0700 |
|---|---|---|
| committer | sniklaus <simon.niklaus@outlook.com> | 2017-09-09 22:59:59 -0700 |
| commit | cb73882b7f6b48f4ba73426b140e77d0d1d97468 (patch) | |
| tree | b2a45d643d3703e489ae2fd18ffd1143b4c7df3e /src/SeparableConvolution_cuda.c | |
no message
Diffstat (limited to 'src/SeparableConvolution_cuda.c')
| -rw-r--r-- | src/SeparableConvolution_cuda.c | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/src/SeparableConvolution_cuda.c b/src/SeparableConvolution_cuda.c new file mode 100644 index 0000000..a48622e --- /dev/null +++ b/src/SeparableConvolution_cuda.c @@ -0,0 +1,23 @@ +#include <THC.h> +#include <THCGeneral.h> + +#include "SeparableConvolution_kernel.h" + +extern THCState* state; + +int SeparableConvolution_cuda_forward( + THCudaTensor* input1, + THCudaTensor* input2, + THCudaTensor* input3, + THCudaTensor* output +) { + SeparableConvolution_kernel_forward( + state, + input1, + input2, + input3, + output + ); + + return 1; +}
\ No newline at end of file |