Changing value based on warp size

ItsPitt · Oct 30, 2023 · 6329d16 · 6329d16
1 parent 8556a27
commit 6329d16
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/faiss/hip/utils/MergeNetworkWarp.h b/faiss/hip/utils/MergeNetworkWarp.h
@@ -164,7 +164,7 @@ template <typename K, typename V, bool Dir, typename Comp, bool Low>
 struct BitonicMergeStep<K, V, 1, Dir, Comp, Low, true> {
     static inline __device__ void merge(K k[1], V v[1]) {
         // Use warp shuffles
-        warpBitonicMergeLE16<K, V, 32, Dir, Comp, true>(k[0], v[0]);
+        warpBitonicMergeLE16<K, V, kWarpSize/2, Dir, Comp, true>(k[0], v[0]);
     }
 };