Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
kernel.cu 334 Bytes
Newer Older
rusty1s's avatar
first cuda kernel try  
rusty1s committed
1
2 
#include <THC/THC.h>
rusty1s's avatar
kernel call needs ifdef cplusplus and header include in cu  
rusty1s committed
3 
#include "kernel.h"
rusty1s's avatar
first cuda kernel try  
rusty1s committed
4
rusty1s's avatar
kernel call check  
rusty1s committed
5
6 
#define scatter_(NAME) TH_CONCAT_4(scatter_, NAME, _kernel_, Real)
#define index_backward TH_CONCAT_2(index_backward_kernel_, Real)
rusty1s's avatar
arg check  
rusty1s committed
7 
#define check TH_CONCAT_2(check_kernel_, Real)
rusty1s's avatar
kernel call needs ifdef cplusplus and header include in cu  
rusty1s committed
8
rusty1s's avatar
max dim  
rusty1s committed
9 
#define MAX_DIMS 25
rusty1s's avatar
added num threads  
rusty1s committed
10 
#define NUM_THREADS 32 * 16
rusty1s's avatar
max dim  
rusty1s committed
11
rusty1s's avatar
kernel call check  
rusty1s committed
12
13 
#include "generic/kernel.cu"
#include "THCGenerateAllTypes.h"