Add support for Volta GPUs. (facebookresearch#517)

Ambssharma · Jul 10, 2018 · 2dc30e1 · 2dc30e1
1 parent a90c36a
commit 2dc30e1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/makefile.inc.in b/makefile.inc.in
@@ -26,6 +26,7 @@ NVCCFLAGS    = -I $(CUDAROOT)/targets/x86_64-linux/include/ \
 -gencode arch=compute_35,code="compute_35" \
 -gencode arch=compute_52,code="compute_52" \
 -gencode arch=compute_60,code="compute_60" \
+-gencode arch=compute_61,code="compute_61" \
 -lineinfo \
 -ccbin $(CXX) -DFAISS_USE_FLOAT16