From 6f98c586bd6e7451646d3b1702b5494ec1d9c530 Mon Sep 17 00:00:00 2001
From: lukec <118525388+sleepcoo@users.noreply.github.com>
Date: Sat, 18 Jan 2025 18:50:37 +0800
Subject: [PATCH] fix sgl-kernel setup.py (#2963)

---
 sgl-kernel/setup.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/sgl-kernel/setup.py b/sgl-kernel/setup.py
index 36596a1b0..33e4abe1b 100644
--- a/sgl-kernel/setup.py
+++ b/sgl-kernel/setup.py
@@ -1,6 +1,6 @@
 from pathlib import Path
 
-from setuptools import setup
+from setuptools import find_packages, setup
 from torch.utils.cpp_extension import BuildExtension, CUDAExtension
 
 root = Path(__file__).parent.resolve()
@@ -67,7 +67,7 @@ ext_modules = [
 setup(
     name="sgl-kernel",
     version=get_version(),
-    packages=["sgl_kernel"],
+    packages=find_packages(),
     package_dir={"": "src"},
     ext_modules=ext_modules,
     cmdclass={"build_ext": BuildExtension},