THCGreedy.h 455 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
#ifndef THC_GREEDY_INC
#define THC_GREEDY_INC

#include <THC/THC.h>

#define THCGreedy_(NAME) TH_CONCAT_4(TH,CReal,Greedy_,NAME)

void THCGreedy_cluster(THCState *state,
                       THCudaLongTensor *cluster,
                       THCudaLongTensor *row,
                       THCudaLongTensor *col,
                       THCudaLongTensor *deg);

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

#endif  // THC_GREEDY_INC