More EIGEN_DEVICE_FUNC fixes for CUDA 10/11/12.

This commit is contained in:
Antonio Sánchez
2023-02-03 19:18:45 +00:00
committed by Rasmus Munk Larsen
parent 2a90653395
commit f6cc359e10
9 changed files with 68 additions and 53 deletions

View File

@@ -191,7 +191,7 @@ template <typename Self, typename Reducer, typename Device,
(TensorEvaluator<typename Self::ChildTypeNoConst, Device>::PacketAccess &&
internal::reducer_traits<Reducer, Device>::PacketAccess)>
struct ScanLauncher {
void operator()(Self& self, typename Self::CoeffReturnType* data) {
void operator()(Self& self, typename Self::CoeffReturnType* data) const {
Index total_size = internal::array_prod(self.dimensions());
// We fix the index along the scan axis to 0 and perform a