Commit 67875011 authored by Boris Mansencal's avatar Boris Mansencal Committed by Matthijs Douze

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

parent cb436146
...@@ -14,7 +14,7 @@ ...@@ -14,7 +14,7 @@
namespace faiss { namespace gpu { namespace faiss { namespace gpu {
#ifdef __CUDA_ARCH__ #ifdef __CUDA_ARCH__
#if __CUDA_ARCH__ <= 600 #if __CUDA_ARCH__ <= 610
constexpr int kWarpSize = 32; constexpr int kWarpSize = 32;
#else #else
#error Unknown __CUDA_ARCH__; please define parameters for compute capability #error Unknown __CUDA_ARCH__; please define parameters for compute capability
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment