diff --git a/.gitmodules b/.gitmodules
index 265d8d9891e..4f9c41ca6d4 100644
--- a/.gitmodules
+++ b/.gitmodules
@@ -1,12 +1,4 @@
-[submodule "sgl-kernel/3rdparty/cutlass"]
-	path = sgl-kernel/3rdparty/cutlass
-	url = https://github.com/NVIDIA/cutlass.git
-[submodule "sgl-kernel/3rdparty/cccl"]
-	path = sgl-kernel/3rdparty/cccl
-	url = https://github.com/NVIDIA/cccl.git
 [submodule "sgl-kernel/3rdparty/flashinfer"]
 	path = sgl-kernel/3rdparty/flashinfer
-	url = https://github.com/flashinfer-ai/flashinfer.git
-[submodule "sgl-kernel/3rdparty/deepgemm"]
-	path = sgl-kernel/3rdparty/deepgemm
-	url = https://github.com/deepseek-ai/DeepGEMM
+	url = https://github.com/sgl-project/flashinfer.git
+	branch = sgl-kernel
diff --git a/sgl-kernel/3rdparty/cccl b/sgl-kernel/3rdparty/cccl
deleted file mode 160000
index b5fe509fd11..00000000000
--- a/sgl-kernel/3rdparty/cccl
+++ /dev/null
@@ -1 +0,0 @@
-Subproject commit b5fe509fd11a925f90d6495176707cc1184eed9d
diff --git a/sgl-kernel/3rdparty/cutlass b/sgl-kernel/3rdparty/cutlass
deleted file mode 160000
index 06e560d98a5..00000000000
--- a/sgl-kernel/3rdparty/cutlass
+++ /dev/null
@@ -1 +0,0 @@
-Subproject commit 06e560d98a5fe8acb975db2c4c26817b6c90acb1
diff --git a/sgl-kernel/3rdparty/deepgemm b/sgl-kernel/3rdparty/deepgemm
deleted file mode 160000
index 3b3783d06cd..00000000000
--- a/sgl-kernel/3rdparty/deepgemm
+++ /dev/null
@@ -1 +0,0 @@
-Subproject commit 3b3783d06cd4d06ac4ba048633e604151d1ee535
diff --git a/sgl-kernel/3rdparty/flashinfer b/sgl-kernel/3rdparty/flashinfer
index 79fd1ae90d9..2b9f16eb79b 160000
--- a/sgl-kernel/3rdparty/flashinfer
+++ b/sgl-kernel/3rdparty/flashinfer
@@ -1 +1 @@
-Subproject commit 79fd1ae90d9b8098ca70dec6071da96f3f6da7b9
+Subproject commit 2b9f16eb79bd344e31725e8d7a92fe7fe980ffdf
diff --git a/sgl-kernel/README.md b/sgl-kernel/README.md
index 90c5b5179dd..3b0c34da826 100644
--- a/sgl-kernel/README.md
+++ b/sgl-kernel/README.md
@@ -36,10 +36,10 @@ docker exec -it sglang_zhyncs /bin/zsh
 
 Third-party libraries:
 
-- [CCCL](https://github.com/NVIDIA/cccl)
 - [CUTLASS](https://github.com/NVIDIA/cutlass)
 - [FlashInfer](https://github.com/flashinfer-ai/flashinfer)
 - [DeepGEMM](https://github.com/deepseek-ai/DeepGEMM)
+- [FlashAttention](https://github.com/Dao-AILab/flash-attention)
 
 ### Kernel Development
 
diff --git a/sgl-kernel/THIRDPARTYNOTICES.txt b/sgl-kernel/THIRDPARTYNOTICES.txt
index e737388f4ea..2a81411ae9e 100644
--- a/sgl-kernel/THIRDPARTYNOTICES.txt
+++ b/sgl-kernel/THIRDPARTYNOTICES.txt
@@ -453,3 +453,36 @@ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 SOFTWARE.
+
+Notice for Dao-AILab/flash-attention
+-------------------------------
+
+BSD 3-Clause License
+
+Copyright (c) 2022, the respective contributors, as shown by the AUTHORS file.
+All rights reserved.
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are met:
+
+* Redistributions of source code must retain the above copyright notice, this
+  list of conditions and the following disclaimer.
+
+* Redistributions in binary form must reproduce the above copyright notice,
+  this list of conditions and the following disclaimer in the documentation
+  and/or other materials provided with the distribution.
+
+* Neither the name of the copyright holder nor the names of its
+  contributors may be used to endorse or promote products derived from
+  this software without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.