fix cpu-only compilation

0fff2951 · Benjamin Thomas Graham · cd155261 · 0fff2951
Commit 0fff2951 authored Apr 01, 2019 by Benjamin Thomas Graham
Show whitespace changes
Inline Side-by-side

Showing with 7 additions and 9 deletions

sparseconvnet/SCN/sparseconvnet_cpu.cpp sparseconvnet/SCN/sparseconvnet_cpu.cpp +7 -9

No files found.
--- a/sparseconvnet/SCN/sparseconvnet_cpu.cpp
+++ b/sparseconvnet/SCN/sparseconvnet_cpu.cpp
@@ -581,13 +581,11 @@ FOO;
 FOO;
 #undef DIMENSION
-at::Tensor CopyFeaturesHelper_updateOutput(at::Tensor rules, at::Tensor context,
+void CopyFeaturesHelper_updateOutput(at::Tensor rules, at::Tensor context,
                                     at::Tensor Context) {
-  return cpu_CopyFeaturesHelper_updateOutput<float>(rules, context, Context);
+  cpu_CopyFeaturesHelper_updateOutput<float>(rules, context, Context);
 }
-at::Tensor CopyFeaturesHelper_updateGradInput(at::Tensor rules,
+void CopyFeaturesHelper_updateGradInput(at::Tensor rules, at::Tensor dcontext,
-                                              at::Tensor dcontext,
                                        at::Tensor dContext) {
-  return cpu_CopyFeaturesHelper_updateGradInput<float>(rules, dcontext,
+  cpu_CopyFeaturesHelper_updateGradInput<float>(rules, dcontext, dContext);
-                                                       dContext);
 }