Commit e6812b97 authored by rusty1s's avatar rusty1s
Browse files

call real kernel

parent 11726e3d
...@@ -20,7 +20,7 @@ void THCTensor_(cumDegree)(THCState *state, THCTensor *self, THCudaLongTensor *i ...@@ -20,7 +20,7 @@ void THCTensor_(cumDegree)(THCState *state, THCTensor *self, THCudaLongTensor *i
real *selfData = THCTensor_(data)(state, self); real *selfData = THCTensor_(data)(state, self);
int64_t *indexData = THCudaLongTensor_data(state, index); int64_t *indexData = THCudaLongTensor_data(state, index);
KERNEL_RUN(cumDegreeKernel, nEdges, selfData, indexData); KERNEL_REAL_RUN(cumDegreeKernel, nEdges, selfData, indexData);
} }
#endif // THC_GENERIC_FILE #endif // THC_GENERIC_FILE
...@@ -5,11 +5,11 @@ ...@@ -5,11 +5,11 @@
void THCTensor_(propose)(THCState *state, THCudaLongTensor *color, THCudaLongTensor *prop, void THCTensor_(propose)(THCState *state, THCudaLongTensor *color, THCudaLongTensor *prop,
THCudaLongTensor *row, THCudaLongTensor *col, THCTensor *weight, THCudaLongTensor *row, THCudaLongTensor *col, THCTensor *weight,
THCudaLongTensor *degree, THCudaLongTensor *cumDegree) { THCudaLongTensor *degree, THCudaLongTensor *cumDegree) {
KERNEL_RUN(weightedProposeKernel, THCudaLongTensor_nElement(state, color), KERNEL_REAL_RUN(weightedProposeKernel, THCudaLongTensor_nElement(state, color),
THCudaLongTensor_data(state, color), THCudaLongTensor_data(state, prop), THCudaLongTensor_data(state, color), THCudaLongTensor_data(state, prop),
THCudaLongTensor_data(state, row), THCudaLongTensor_data(state, col), THCudaLongTensor_data(state, row), THCudaLongTensor_data(state, col),
THCTensor_(data)(state, weight), THCudaLongTensor_data(state, degree), THCTensor_(data)(state, weight), THCudaLongTensor_data(state, degree),
THCudaLongTensor_data(state, cumDegree)); THCudaLongTensor_data(state, cumDegree));
} }
#endif // THC_GENERIC_FILE #endif // THC_GENERIC_FILE
......
...@@ -5,11 +5,11 @@ ...@@ -5,11 +5,11 @@
void THCTensor_(response)(THCState *state, THCudaLongTensor *color, THCudaLongTensor *prop, void THCTensor_(response)(THCState *state, THCudaLongTensor *color, THCudaLongTensor *prop,
THCudaLongTensor *row, THCudaLongTensor *col, THCTensor *weight, THCudaLongTensor *row, THCudaLongTensor *col, THCTensor *weight,
THCudaLongTensor *degree, THCudaLongTensor *cumDegree) { THCudaLongTensor *degree, THCudaLongTensor *cumDegree) {
KERNEL_RUN(weightedResponseKernel, THCudaLongTensor_nElement(state, color), KERNEL_REAL_RUN(weightedResponseKernel, THCudaLongTensor_nElement(state, color),
THCudaLongTensor_data(state, color), THCudaLongTensor_data(state, prop), THCudaLongTensor_data(state, color), THCudaLongTensor_data(state, prop),
THCudaLongTensor_data(state, row), THCudaLongTensor_data(state, col), THCudaLongTensor_data(state, row), THCudaLongTensor_data(state, col),
THCTensor_(data)(state, weight), THCudaLongTensor_data(state, degree), THCTensor_(data)(state, weight), THCudaLongTensor_data(state, degree),
THCudaLongTensor_data(state, cumDegree)); THCudaLongTensor_data(state, cumDegree));
} }
#endif // THC_GENERIC_FILE #endif // THC_GENERIC_FILE
...@@ -4,7 +4,7 @@ ...@@ -4,7 +4,7 @@
void THCCTensor_(grid)(THCudaLongTensor *self, THCTensor *pos, THCTensor *size, void THCCTensor_(grid)(THCudaLongTensor *self, THCTensor *pos, THCTensor *size,
THCudaLongTensor *count) { THCudaLongTensor *count) {
/* THCTensor_(grid)(state, self, pos, size, count); */ THCTensor_(grid)(state, self, pos, size, count);
} }
#endif #endif
......
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