add support for Nvidia GPUs with CUDA compute capability 6.1 (#110)

67875011 · Boris Mansencal · Matthijs Douze · cb436146 · 67875011
Commit 67875011 authored May 31, 2017 by Boris Mansencal Committed by Matthijs Douze May 31, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

DeviceDefs.cuh gpu/utils/DeviceDefs.cuh +1 -1

No files found.
--- a/gpu/utils/DeviceDefs.cuh
+++ b/gpu/utils/DeviceDefs.cuh
@@ -14,7 +14,7 @@
 namespace faiss { namespace gpu {

 #ifdef __CUDA_ARCH__
-#if __CUDA_ARCH__ <= 600
+#if __CUDA_ARCH__ <= 610
 constexpr int kWarpSize = 32;
 #else
 #error Unknown __CUDA_ARCH__; please define parameters for compute capability