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

fix sgl-kernel setup.py (#2963)

parent 656dcc1a
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},
......
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