Fixed iterator diff for thrust iterator adapter

STEllAR-GROUP · Nov 18, 2019 · 35ca5ec · 35ca5ec
1 parent 4618370
commit 35ca5ec
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/blaze_cuda/util/algorithms/CUDATransform.h b/blaze_cuda/util/algorithms/CUDATransform.h
@@ -85,12 +85,16 @@ class ThrustInputIteratorAdapter
    BLAZE_ALWAYS_INLINE BLAZE_DEVICE_CALLABLE auto
       operator-( ThrustInputIteratorAdapter const& other ) const noexcept
    {
-      return ThrustInputIteratorAdapter( it - other.it );
+      return it - other.it;
    }
 
    BLAZE_ALWAYS_INLINE BLAZE_DEVICE_CALLABLE auto operator+( ptrdiff_t inc ) const noexcept {
       return ThrustInputIteratorAdapter( IteratorType( it + inc ) );
    }
+
+   BLAZE_ALWAYS_INLINE BLAZE_DEVICE_CALLABLE auto operator-( ptrdiff_t dec ) const noexcept {
+      return ThrustInputIteratorAdapter( IteratorType( it - dec ) );
+   }
 };
 
 template< typename IteratorType >