Dororo99/Ours_S3GS_Waymo

1#2# Copyright (C) 2023, Inria3# GRAPHDECO research group, https://team.inria.fr/graphdeco4# All rights reserved.5#6# This software is free for non-commercial, research and evaluation use 7# under the terms of the LICENSE.md file.8#9# For inquiries contact  george.drettakis@inria.fr10#11 12from setuptools import setup13from torch.utils.cpp_extension import CUDAExtension, BuildExtension14import os15os.path.dirname(os.path.abspath(__file__))16 17setup(18    name="diff_gaussian_rasterization",19    packages=['diff_gaussian_rasterization'],20    ext_modules=[21        CUDAExtension(22            name="diff_gaussian_rasterization._C",23            sources=[24            "cuda_rasterizer/rasterizer_impl.cu",25            "cuda_rasterizer/forward.cu",26            "cuda_rasterizer/backward.cu",27            "rasterize_points.cu",28            "ext.cpp"],29            extra_compile_args={"nvcc": ["-I" + os.path.join(os.path.dirname(os.path.abspath(__file__)), "third_party/glm/")]})30        ],31    cmdclass={32        'build_ext': BuildExtension33    }34)35