From a482f58623e78b30784bb087856ff5b184178ef8 Mon Sep 17 00:00:00 2001
From: Lucas Wilkinson <lwilkinson@neuralmagic.com>
Date: Thu, 23 Jan 2025 23:16:31 +0000
Subject: [PATCH] update fa location

Signed-off-by: Lucas Wilkinson <lwilkinson@neuralmagic.com>
---
 CMakeLists.txt | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index e094804df9473..2f9da6fa3e1d3 100755
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -575,8 +575,8 @@ if(VLLM_FLASH_ATTN_SRC_DIR)
 else()
   FetchContent_Declare(
           vllm-flash-attn
-          GIT_REPOSITORY https://github.com/neuralmagic/vllm-flash-attention.git
-          GIT_TAG 28e69b72483f007432acf403d752eae51f7df4c2
+          GIT_REPOSITORY https://github.com/vllm-project/flash-attention.git
+          GIT_TAG 0aff05f577e8a10086066a00618609199b25231d
           GIT_PROGRESS TRUE
           # Don't share the vllm-flash-attn build between build types
           BINARY_DIR ${CMAKE_BINARY_DIR}/vllm-flash-attn