THCDegree.cu 393 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
void THCDegree(THCState *state, THCudaLongTensor *self, THCudaLongTensor *index) {
  int nEdges = THCudaLongTensor_nElement(state, index);
  THCudaLongTensor *one = THCudaLongTensor_newWithSize1d(state, nEdges);
  THCudaLongTensor_fill(state, one, 1);

  THCudaLongTensor_fill(state, self, 0);
  THCudaLongTensor_scatterAdd(state, self, 0, index, one);

  THCudaLongTensor_free(state, one);
}