THCGreedy.cu 707 Bytes
Newer Older
rusty1s's avatar
rusty1s committed
1
2
#include "THCGreedy.h"

rusty1s's avatar
rusty1s committed
3
4
#include "THCColor.cu"

rusty1s's avatar
rusty1s committed
5
6
void THCGreedy(THCState *state, THCudaLongTensor *cluster, THCudaLongTensor *row,
               THCudaLongTensor *col, THCudaLongTensor *deg) {
rusty1s's avatar
rusty1s committed
7
8
9
10
11
12
13
14
  THCAssertSameGPU(THCudaLongTensor_checkGPU(state, 4, cluster, row, col, deg));

  THCudaLongTensor_fill(state, cluster, -1);
  THCGreedy_assignColor(state, cluster);
  /* while(!THCGreedy_assignColor(state, cluster)) { */
  /*   printf("DRIN"); */
  /* }; */

rusty1s's avatar
rusty1s committed
15
16
17
18
19
20
21
22
23
  // Fill cluster with -1
  // assign color to clusters < 0 (return done)
  // Generate proposal vector with length of nodes (init?)
  // call propose step
  // call response step
}

#include "generic/THCGreedy.cu"
#include "THC/THCGenerateAllTypes.h"