未验证 提交 30ecd17c 编写于 作者: X XuanYang-cn 提交者: GitHub

fix faiss cannot build gpu bug (#3424)

Signed-off-by: Nyangxuan <xuan.yang@zilliz.com>
Co-authored-by: Nyangxuan <xuan.yang@zilliz.com>
上级 ff9fe6fa
......@@ -553,7 +553,7 @@ macro(build_faiss)
endif ()
if (MILVUS_GPU_VERSION)
if (MILVUS_CUDA_ARCH STREQUAL "DEFAULT")
if (NOT MILVUS_CUDA_ARCH OR MILVUS_CUDA_ARCH STREQUAL "DEFAULT")
set(FAISS_CONFIGURE_ARGS ${FAISS_CONFIGURE_ARGS}
"--with-cuda=${CUDA_TOOLKIT_ROOT_DIR}"
"--with-cuda-arch=-gencode=arch=compute_60,code=sm_60 -gencode=arch=compute_61,code=sm_61 -gencode=arch=compute_70,code=sm_70 -gencode=arch=compute_75,code=sm_75"
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册