mirrors
/
faiss
mirror of https://github.com/facebookresearch/faiss.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Update kWarpSize for A100 arch (#1380) 

Summary:
Update define for 8.0 compute capability.

Pull Request resolved: https://github.com/facebookresearch/faiss/pull/1380

Test Plan: A100 builds on Google Cloud.

Reviewed By: mdouze

Differential Revision: D23357243

Pulled By: wickedfoo

fbshipit-source-id: 5c64ecc93bfe0bfa853e9d12def15c6bbe4c5c0e
pull/1382/head
Jeff Johnson 2020-08-27 10:18:03 -07:00 committed by Facebook GitHub Bot
parent 8941160678
commit 57506c6f4d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
faiss/gpu/utils/DeviceDefs.cuh
View File

@ -18,7 +18,7 @@ namespace faiss { namespace gpu {
#endif
#ifdef __CUDA_ARCH__
#if __CUDA_ARCH__ <= 750
#if __CUDA_ARCH__ <= 800
constexpr int kWarpSize = 32;
#else
#error Unknown __CUDA_ARCH__; please define parameters for compute capability