diff options
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 |