Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
kernel.cu 217 Bytes
Newer Older
rusty1s's avatar
first cuda kernel try  
rusty1s committed
1
2
3
4
5
6
7 
#include <THC/THC.h>

#define scatter_(NAME) TH_CONCAT_4(scatter_, NAME, _kernel_, Real)
#define index_backward TH_CONCAT_2(index_backward_kernel_, Real)

#include "generic/kernel.cu"
#include "THCGenerateAllTypes.h"