Skip to content

GitLab

Unverified Commit 6f98c586 authored by lukec's avatar lukec Committed by GitHub
Browse files

fix sgl-kernel setup.py (#2963)

parent 656dcc1a
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
sgl-kernel/setup.py
View file @ 6f98c586
from pathlib import Path from pathlib import Path
from setuptools import setup from setuptools import find_packages, setup
from torch.utils.cpp_extension import BuildExtension, CUDAExtension from torch.utils.cpp_extension import BuildExtension, CUDAExtension
root = Path(__file__).parent.resolve() root = Path(__file__).parent.resolve()
...@@ -67,7 +67,7 @@ ext_modules = [ ...@@ -67,7 +67,7 @@ ext_modules = [
setup( setup(
name="sgl-kernel", name="sgl-kernel",
version=get_version(), version=get_version(),
packages=["sgl_kernel"], packages=find_packages(),
package_dir={"": "src"}, package_dir={"": "src"},
ext_modules=ext_modules, ext_modules=ext_modules,
cmdclass={"build_ext": BuildExtension}, cmdclass={"build_ext": BuildExtension},
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment