Make it optional to build CUDA extension for SAM 2; also fallback to all available kernels if Flash Attention fails#155

Merged

ronghanghu merged 1 commit intofacebookresearch:mainfrom ronghanghu:ronghanghu/cuda_kernel_optional

Aug 6, 2024

+173-33

Commits on Aug 6, 2024

Make it optional to build CUDA extension for SAM 2; also fallback to all available kernels if Flash Attention fails
ronghanghu
committed