Added missing define

2026-04-10 11:34:33 +08:00 · 2015-12-22 16:11:07 -08:00
parent e7e6d01810
commit 9c7d96697b
1 changed files with 3 additions and 0 deletions
--- a/unsupported/Eigen/CXX11/src/Tensor/TensorReductionCuda.h
+++ b/unsupported/Eigen/CXX11/src/Tensor/TensorReductionCuda.h
@@ -131,6 +131,7 @@ struct FullReducer<Self, Op, GpuDevice, Vectorizable> {
  }
 };

+#define DIVUP(x, y) (((x) + (y)-1) / (y))

 template <int NumPerThread, typename Self,
          typename Reducer, typename Index>
@@ -185,6 +186,8 @@ struct OuterReducer<Self, Op, GpuDevice> {
  }
 };

+#undef DIVUP
+
 #endif