added num threads

torch_scatter/kernel/generic/kernel.cu

@@ -10,6 +10,9 @@ void check(THCState *state, THCTensor *output, THCudaLongTensor *index, THCTenso
 
 void scatter_(mul)(THCState *state, int dim, THCTensor *output, THCudaLongTensor *index, THCTensor *input) {
   check(state, output, index, input);
+
+  const ptrdiff_t n = THCudaLongTensor_nElement(state, index);
+  const dim3 block = dim3(NUM_THREADS);
   printf("mul");
 }
 

torch_scatter/kernel/kernel.cu

@@ -7,6 +7,7 @@
 #define check TH_CONCAT_2(check_kernel_, Real)
 
 #define MAX_DIMS 25
+#define NUM_THREADS 32 * 16
 
 #include "generic/kernel.cu"
 #include "THCGenerateAllTypes.h"