Commit 51a71064 authored by rusty1s's avatar rusty1s
Browse files

thcc wrapper methods

parent af9b769a
This directory contains the low-level tensor libraries.
There are multiple variants of the library, summarized here:
* TH = TorcH
* THC = TorcH Cuda
* THCC = TorcH Cuda Connect
#include <THC/THC.h>
#include "THCGreedy.h"
#define THCCGreedy_ TH_CONCAT_3(THCC,Real,Greedy)
#define THCGreedy_ TH_CONCAT_3(THC,CReal,Greedy)
extern THCState *state;
void THCCGreedy(THCudaLongTensor *cluster, THCudaLongTensor *row, THCudaLongTensor *col,
THCudaLongTensor *deg) {
THCGreedy(state, cluster, row, col, deg);
}
#include "generic/THCCGreedy.c"
#include "THCGenerateAllTypes.h"
void THCCGreedy(THCudaLongTensor *cluster, THCudaLongTensor *row, THCudaLongTensor *col,
THCudaLongTensor *deg);
void THCCByteGreedy(THCudaLongTensor *cluster, THCudaLongTensor *row, THCudaLongTensor *col,
THCudaLongTensor *deg, THCudaByteTensor *weight);
void THCCCharGreedy(THCudaLongTensor *cluster, THCudaLongTensor *row, THCudaLongTensor *col,
THCudaLongTensor *deg, THCudaCharTensor *weight);
void THCCShortGreedy(THCudaLongTensor *cluster, THCudaLongTensor *row, THCudaLongTensor *col,
THCudaLongTensor *deg, THCudaShortTensor *weight);
void THCCIntGreedy(THCudaLongTensor *cluster, THCudaLongTensor *row, THCudaLongTensor *col,
THCudaLongTensor *deg, THCudaIntTensor *weight);
void THCCLongGreedy(THCudaLongTensor *cluster, THCudaLongTensor *row, THCudaLongTensor *col,
THCudaLongTensor *deg, THCudaLongTensor *weight);
void THCCFloatGreedy(THCudaLongTensor *cluster, THCudaLongTensor *row, THCudaLongTensor *col,
THCudaLongTensor *deg, THCudaTensor *weight);
void THCCDoubleGreedy(THCudaLongTensor *cluster, THCudaLongTensor *row, THCudaLongTensor *col,
THCudaLongTensor *deg, THCudaDoubleTensor *weight);
#include <THC/THC.h>
#include "THCGrid.h"
#define THCCGrid_ TH_CONCAT_3(THCC,Real,Grid)
#define THCGrid_ TH_CONCAT_3(THC,CReal,Grid)
extern THCState *state;
#include "generic/THCCGrid.c"
#include "THCGenerateAllTypes.h"
void THCCByteGrid(THCudaLongTensor *cluster, THCudaByteTensor *pos, THCudaByteTensor *size,
THCudaLongTensor *count);
void THCCCharGrid(THCudaLongTensor *cluster, THCudaCharTensor *pos, THCudaCharTensor *size,
THCudaLongTensor *count);
void THCCShortGrid(THCudaLongTensor *cluster, THCudaShortTensor *pos, THCudaShortTensor *size,
THCudaLongTensor *count);
void THCCIntGrid(THCudaLongTensor *cluster, THCudaIntTensor *pos, THCudaIntTensor *size,
THCudaLongTensor *count);
void THCCLongGrid(THCudaLongTensor *cluster, THCudaLongTensor *pos, THCudaLongTensor *size,
THCudaLongTensor *count);
void THCCFloatGrid(THCudaLongTensor *cluster, THCudaTensor *pos, THCudaTensor *size,
THCudaLongTensor *count);
void THCCDoubleGrid(THCudaLongTensor *cluster, THCudaDoubleTensor *pos, THCudaDoubleTensor *size,
THCudaLongTensor *count);
#ifndef THC_GENERIC_FILE
#define THC_GENERIC_FILE "generic/THCCGreedy.c"
#else
void THCCGreedy_(THCudaLongTensor *cluster, THCudaLongTensor *row, THCudaLongTensor *col,
THCudaLongTensor *deg, THCTensor *weight) {
THCGreedy_(state, cluster, row, col, deg, weight);
}
#endif
#ifndef THC_GENERIC_FILE
#define THC_GENERIC_FILE "generic/THCCGrid.c"
#else
void THCCGrid_(THCudaLongTensor *cluster, THCTensor *pos, THCTensor *size,
THCudaLongTensor *count) {
THCGrid_(state, cluster, pos, size, count);
}
#endif
void THCudaGreedyWrapper(THCudaLongTensor *cluster,
THCudaLongTensor *row,
THCudaLongTensor *col,
THCudaLongTensor *deg);
void THCudaByteGreedyWrapper(THCudaLongTensor *cluster,
THCudaLongTensor *row,
THCudaLongTensor *col,
THCudaLongTensor *deg,
THCudaByteTensor *weight);
void THCudaCharGreedyWrapper(THCudaLongTensor *cluster,
THCudaLongTensor *row,
THCudaLongTensor *col,
THCudaLongTensor *deg,
THCudaCharTensor *weight);
void THCudaShortGreedyWrapper(THCudaLongTensor *cluster,
THCudaLongTensor *row,
THCudaLongTensor *col,
THCudaLongTensor *deg,
THCudaShortTensor *weight);
void THCudaIntGreedyWrapper(THCudaLongTensor *cluster,
THCudaLongTensor *row,
THCudaLongTensor *col,
THCudaLongTensor *deg,
THCudaIntTensor *weight);
void THCudaLongGreedyWrapper(THCudaLongTensor *cluster,
THCudaLongTensor *row,
THCudaLongTensor *col,
THCudaLongTensor *deg,
THCudaLongTensor *weight);
void THCudaFloatGreedyWrapper(THCudaLongTensor *cluster,
THCudaLongTensor *row,
THCudaLongTensor *col,
THCudaLongTensor *deg,
THCudaTensor *weight);
void THCudaDoubleGreedyWrapper(THCudaLongTensor *cluster,
THCudaLongTensor *row,
THCudaLongTensor *col,
THCudaLongTensor *deg,
THCudaDoubleTensor *weight);
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment