"vscode:/vscode.git/clone" did not exist on "07c0fe4b87a07fc1b42bac738f013b78833559ae"
Commit d48533ea authored by rusty1s's avatar rusty1s
Browse files

added conv kernel headers

parent de4f89b7
#include "THCBasis.cu" #include "THCBasis.cu"
#include "THCConv.cu"
...@@ -2,6 +2,7 @@ ...@@ -2,6 +2,7 @@
#define THC_INC #define THC_INC
#include "THCBasis.h" #include "THCBasis.h"
#include "THCConv.h"
#endif // THC_INC #endif // THC_INC
#include "THCConv.h"
#include "generic/THCConv.cu"
#include "THC/THCGenerateFloatTypes.h"
#ifndef THC_CONV_INC
#define THC_CONV_INC
#include <THC/THC.h>
#ifdef __cplusplus
extern "C" {
#endif // __cplusplus
#include "generic/THCConv.h"
#include "THC/THCGenerateFloatTypes.h"
#ifdef __cplusplus
}
#endif // __cplusplus
#endif // THC_CONV_INC
#ifndef THC_GENERIC_FILE
#define THC_GENERIC_FILE "generic/THCConv.cu"
#else
void THCTensor_(convForward)(THCState *state, THCTensor *self, THCTensor *src, THCTensor *weight,
THCTensor *basis, THCudaLongTensor *weightIndex) {
}
void THCTensor_(convBackwardSrc)(THCState *state, THCTensor *self, THCTensor *gradOutput,
THCTensor *weight, THCTensor *basis,
THCudaLongTensor *weightIndex) {
}
void THCTensor_(convBackwardBasis)(THCState *state, THCTensor *self, THCTensor *gradOutput,
THCTensor *src, THCTensor *weight,
THCudaLongTensor *weightIndex) {
}
void THCTensor_(convBackwardWeight)(THCState *state, THCTensor *self, THCTensor *gradOutput,
THCTensor *src, THCTensor *basis,
THCudaLongTensor *weightIndex) {
}
#endif // THC_GENERIC_FILE
#ifndef THC_GENERIC_FILE
#define THC_GENERIC_FILE "generic/THCConv.h"
#else
void THCTensor_(convForward)(THCState *state, THCTensor *self, THCTensor *src, THCTensor *weight,
THCTensor *basis, THCudaLongTensor *weightIndex);
void THCTensor_(convBackwardSrc)(THCState *state, THCTensor *self, THCTensor *gradOutput,
THCTensor *weight, THCTensor *basis,
THCudaLongTensor *weightIndex);
void THCTensor_(convBackwardBasis)(THCState *state, THCTensor *self, THCTensor *gradOutput,
THCTensor *src, THCTensor *weight,
THCudaLongTensor *weightIndex);
void THCTensor_(convBackwardWeight)(THCState *state, THCTensor *self, THCTensor *gradOutput,
THCTensor *src, THCTensor *basis,
THCudaLongTensor *weightIndex);
#endif // THC_GENERIC_FILE
...@@ -4,18 +4,22 @@ ...@@ -4,18 +4,22 @@
void THCCTensor_(convForward)(THCTensor *self, THCTensor *src, THCTensor *weight, THCTensor *basis, void THCCTensor_(convForward)(THCTensor *self, THCTensor *src, THCTensor *weight, THCTensor *basis,
THCudaLongTensor *weightIndex) { THCudaLongTensor *weightIndex) {
THCTensor_(convForward)(state, self, src, weight, basis, weightIndex);
} }
void THCCTensor_(convBackwardSrc)(THCTensor *self, THCTensor *gradOutput, THCTensor *weight, void THCCTensor_(convBackwardSrc)(THCTensor *self, THCTensor *gradOutput, THCTensor *weight,
THCTensor *basis, THCudaLongTensor *weightIndex) { THCTensor *basis, THCudaLongTensor *weightIndex) {
THCTensor_(convBackwardSrc)(state, self, gradOutput, weight, basis, weightIndex);
} }
void THCCTensor_(convBackwardBasis)(THCTensor *self, THCTensor *gradOutput, THCTensor *src, void THCCTensor_(convBackwardBasis)(THCTensor *self, THCTensor *gradOutput, THCTensor *src,
THCTensor *weight, THCudaLongTensor *weightIndex) { THCTensor *weight, THCudaLongTensor *weightIndex) {
THCTensor_(convBackwardBasis)(state, self, gradOutput, src, weight, weightIndex);
} }
void THCCTensor_(convBackwardWeight)(THCTensor *self, THCTensor *gradOutput, THCTensor *src, void THCCTensor_(convBackwardWeight)(THCTensor *self, THCTensor *gradOutput, THCTensor *src,
THCTensor *basis, THCudaLongTensor *weightIndex) { THCTensor *basis, THCudaLongTensor *weightIndex) {
THCTensor_(convBackwardWeight)(state, self, gradOutput, src, basis, weightIndex);
} }
#endif // THC_GENERIC_FILE #endif // THC_GENERIC_FILE
......
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