THCResponse.cuh 791 Bytes
Newer Older
rusty1s's avatar
rusty1s committed
1
2
3
4
5
6
7
#ifndef THC_GENERIC_FILE
#define THC_GENERIC_FILE "generic/THCResponse.cuh"
#else

void THCTensor_(response)(THCState *state, THCudaLongTensor *color, THCudaLongTensor *prop,
                          THCudaLongTensor *row, THCudaLongTensor *col, THCTensor *weight,
                          THCudaLongTensor *degree, THCudaLongTensor *cumDegree) {
rusty1s's avatar
rusty1s committed
8
9
10
11
12
  KERNEL_REAL_RUN(weightedResponseKernel, THCudaLongTensor_nElement(state, color),
                  THCudaLongTensor_data(state, color), THCudaLongTensor_data(state, prop),
                  THCudaLongTensor_data(state, row), THCudaLongTensor_data(state, col),
                  THCTensor_(data)(state, weight), THCudaLongTensor_data(state, degree),
                  THCudaLongTensor_data(state, cumDegree));
rusty1s's avatar
rusty1s committed
13
14
15
}

#endif  // THC_GENERIC_FILE