Fixing a warp 64 issue

ItsPitt · Oct 30, 2023 · 8556a27 · 8556a27
1 parent 3284dfd
commit 8556a27
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, 16, Dir, Comp, true>(k[0], v[0]);
+        warpBitonicMergeLE16<K, V, 32, Dir, Comp, true>(k[0], v[0]);
     }
 };