ShorterSplatting/setup.py at main · MachinePerceptionLab/ShorterSplatting · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
from setuptools import setup, find_packages

def get_cuda_modules():
    from torch.utils.cpp_extension import CUDAExtension
    ext_modules = [
        CUDAExtension(
            name="fused_ssim_cuda",
            sources=[
            "litegs/submodules/fused_ssim/ssim.cu",
            "litegs/submodules/fused_ssim/ext.cpp"]),
        CUDAExtension(
                name="simple_knn._C",
                sources=[
                "litegs/submodules/simple-knn/spatial.cu",
                "litegs/submodules/simple-knn/simple_knn.cu",
                "litegs/submodules/simple-knn/ext.cpp"]),
        CUDAExtension(
                name="litegs_fused",
                sources=[
                "litegs/submodules/gaussian_raster/binning.cu",
                "litegs/submodules/gaussian_raster/compact.cu",
                "litegs/submodules/gaussian_raster/cuda_errchk.cpp",
                "litegs/submodules/gaussian_raster/ext_cuda.cpp",
                "litegs/submodules/gaussian_raster/raster.cu",
                "litegs/submodules/gaussian_raster/transform.cu"])
    ]
    return ext_modules

def get_cmdclass():
    from torch.utils.cpp_extension import BuildExtension
    return {'build_ext': BuildExtension}

setup(
    name="lite-gaussian-splatting",
    version="1.0.0",
    author="Kaimin Liao",
    author_email="kaiminliao@gmail.com",
    description="A High-Performance Modular Framework for Gaussian Splatting Training",
    long_description_content_type="text/markdown",
    url="https://github.com/MooreThreads/LiteGS",
    packages=find_packages(where=".", include=["litegs", "litegs.*"])+["fused_ssim"],
    package_dir={"litegs": "litegs","fused_ssim":"litegs/submodules/fused_ssim/fused_ssim"},
    setup_requires=["torch","wheel"],
    install_requires=["torch","wheel","numpy","plyfile","tqdm","pillow"],
    ext_modules=get_cuda_modules(),
    include_package_data=True,
    cmdclass=get_cmdclass(),
    python_requires=">=3.8",
    classifiers=[
        "Programming Language :: Python :: 3",
        "Operating System :: OS Independent",
    ],
)