fix sgl-kernel setup.py (#2963)
This commit is contained in:
@@ -1,6 +1,6 @@
|
|||||||
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 = [
|
|||||||
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},
|
||||||
|
|||||||
Reference in New Issue
Block a user