From 389b81be130c175b59541e2b7a10bc29c6817f81 Mon Sep 17 00:00:00 2001
From: Georgy Evtushenko <evtushenko.georgy@gmail.com>
Date: Wed, 25 May 2022 15:48:21 +0400
Subject: [PATCH] Fix in-place execution of adjacent difference

---
 cub/device/dispatch/dispatch_adjacent_difference.cuh | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/cub/device/dispatch/dispatch_adjacent_difference.cuh b/cub/device/dispatch/dispatch_adjacent_difference.cuh
index 5438dcccd7..916db5463b 100644
--- a/cub/device/dispatch/dispatch_adjacent_difference.cuh
+++ b/cub/device/dispatch/dispatch_adjacent_difference.cuh
@@ -106,7 +106,7 @@ DeviceAdjacentDifferenceDifferenceKernel(InputIteratorT input,
   agent.Process(tile_idx, tile_base);
 }
 
-template <typename InputIteratorT>
+template <typename InputIteratorT, bool MayAlias = true>
 struct DeviceAdjacentDifferencePolicy
 {
   using ValueT = typename std::iterator_traits<InputIteratorT>::value_type;
@@ -131,7 +131,7 @@ struct DeviceAdjacentDifferencePolicy
       AgentAdjacentDifferencePolicy<128,
                                     Nominal8BItemsToItems<ValueT>(7),
                                     BLOCK_LOAD_WARP_TRANSPOSE,
-                                    LOAD_LDG,
+                                    MayAlias ? LOAD_CA : LOAD_LDG,
                                     BLOCK_STORE_WARP_TRANSPOSE>;
   };
 
@@ -145,7 +145,7 @@ template <typename InputIteratorT,
           bool MayAlias,
           bool ReadLeft,
           typename SelectedPolicy =
-            DeviceAdjacentDifferencePolicy<InputIteratorT>>
+            DeviceAdjacentDifferencePolicy<InputIteratorT, MayAlias>>
 struct DispatchAdjacentDifference : public SelectedPolicy
 {
   using InputT = typename std::iterator_traits<InputIteratorT>::value_type;