Add support for Volta GPUs. (#517)

2dc30e14 · Lucas Hosseini · GitHub · a90c36a3 · 2dc30e14
Unverified Commit 2dc30e14 authored Jul 10, 2018 by Lucas Hosseini Committed by GitHub Jul 10, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

makefile.inc.in makefile.inc.in +1 -0

No files found.
--- 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