THCResponse.cuh 766 Bytes
Newer Older
rusty1s's avatar
rusty1s committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
#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) {
  KERNEL_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));
}

#endif  // THC_GENERIC_FILE