THCGraclus.cu 613 Bytes
Newer Older
rusty1s's avatar
rusty1s committed
1
2
3
4
5
6
#ifndef THC_GENERIC_FILE
#define THC_GENERIC_FILE "generic/THCGraclus.cu"
#else

void THCTensor_(graclus)(THCState *state, THCudaLongTensor *self, THCudaLongTensor *row,
                         THCudaLongTensor *col, THCTensor *weight) {
rusty1s's avatar
rusty1s committed
7
8
9
  THCAssertSameGPU(THCTensor_(checkGPU)(state, 4, self, row, col, weight));

  THC_TENSOR_GRACLUS(state, self, row,
rusty1s's avatar
rusty1s committed
10
    while(!THCudaLongTensor_color(state, self)) {
rusty1s's avatar
rusty1s committed
11
12
      THCTensor_(propose)(state, self, prop, row, col, weight, degree, cumDegree);
      THCTensor_(response)(state, self, prop, row, col, weight, degree, cumDegree);
rusty1s's avatar
rusty1s committed
13
    }
rusty1s's avatar
rusty1s committed
14
  )
rusty1s's avatar
rusty1s committed
15
16
17
}

#endif  // THC_GENERIC_FILE