Skip to content

GitLab

Commit 6997baa9 authored by rusty1s's avatar rusty1s
Browse files

bugfixes

parent a9209b48
Hide whitespace changes
Inline Side-by-side
Showing with 16 additions and 0 deletions
aten/THC/THCGreedy.h
View file @ 6997baa9
...@@ -5,10 +5,18 @@ ...@@ -5,10 +5,18 @@
#define THCGreedy_ TH_CONCAT_3(TH,CReal,Greedy) #define THCGreedy_ TH_CONCAT_3(TH,CReal,Greedy)
#ifdef __cplusplus
extern "C" {
#endif // __cplusplus
void THCGreedy(THCState *state, THCudaLongTensor *cluster, THCudaLongTensor *row, void THCGreedy(THCState *state, THCudaLongTensor *cluster, THCudaLongTensor *row,
THCudaLongTensor *col, THCudaLongTensor *deg); THCudaLongTensor *col, THCudaLongTensor *deg);
#include "generic/THCGreedy.h" #include "generic/THCGreedy.h"
#include "THC/THCGenerateAllTypes.h" #include "THC/THCGenerateAllTypes.h"
#ifdef __cplusplus
}
#endif // __cplusplus
#endif // THC_GREEDY_INC #endif // THC_GREEDY_INC
aten/THC/THCGrid.h
View file @ 6997baa9
...@@ -5,7 +5,15 @@ ...@@ -5,7 +5,15 @@
#define THCGrid_ TH_CONCAT_3(TH,CReal,Grid) #define THCGrid_ TH_CONCAT_3(TH,CReal,Grid)
#ifdef __cplusplus
extern "C" {
#endif // __cplusplus
#include "generic/THCGrid.h" #include "generic/THCGrid.h"
#include "THC/THCGenerateAllTypes.h" #include "THC/THCGenerateAllTypes.h"
#ifdef __cplusplus
}
#endif // __cplusplus
#endif // THC_GRID_INC #endif // THC_GRID_INC
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