Workaround for CUDA 11.4.4 build in Conda on Ubuntu 22 / v6 kernel

Summary: When building with CUDA 11.4.4, CMake does not properly include files under Conda environment. This workaround flattens the include sub-directories in to the include root. It will unblock us for now while we are looking for a fix through CMakeLists files or figure out if it's a CMake bug and it gets fixed.

Differential Revision: D57545169

Author: Ramil Bakhshyiev

conda/faiss-gpu/build-lib.sh

@@ -6,6 +6,12 @@
 
 set -e
 
+# Workaround for CUDA 11.4.4 builds. Moves all necessary headers to include root.
+if [[ -n "$FAISS_FLATTEN_CONDA_INCLUDES" && "$FAISS_FLATTEN_CONDA_INCLUDES" == "1" ]]; then
+  cp -r -n $CONDA_PREFIX/x86_64-conda-linux-gnu/sysroot/usr/include/* $CONDA_PREFIX/include/
+  cp -r -n $CONDA_PREFIX/x86_64-conda-linux-gnu/include/c++/11.2.0/* $CONDA_PREFIX/include/
+  cp -r -n $CONDA_PREFIX/x86_64-conda-linux-gnu/include/c++/11.2.0/x86_64-conda-linux-gnu/* $CONDA_PREFIX/include/
+fi
 
 # Build libfaiss.so/libfaiss_avx2.so/libfaiss_avx512.so
 cmake -B _build \

conda/faiss-gpu/meta.yaml

@@ -43,6 +43,9 @@ outputs:
         - {{ pin_compatible('libfaiss', exact=True) }}
       script_env:
         - CUDA_ARCHS
+        {% if cudatoolkit == '11.4.4' %}
+        - FAISS_FLATTEN_CONDA_INCLUDES=1
+        {% endif %}
     requirements:
       build:
         - {{ compiler('cxx') }}