THCGreedy.cu 542 Bytes
Newer Older
rusty1s's avatar
rusty1s committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
#include "THCGreedy.h"

void THCGreedy_cluster(THCState *state,
                       THCudaLongTensor *cluster,
                       THCudaLongTensor *row,
                       THCudaLongTensor *col,
                       THCudaLongTensor *deg) {
  printf("THCGreedy_cluster drin");
  // 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"