souging/TRELLIS_TextTo3D

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="vox2seq",19    packages=['vox2seq', 'vox2seq.pytorch'],20    ext_modules=[21        CUDAExtension(22            name="vox2seq._C",23            sources=[24                "src/api.cu",25                "src/z_order.cu",26                "src/hilbert.cu",27                "src/ext.cpp",28            ],29        )30    ],31    cmdclass={32        'build_ext': BuildExtension33    }34)35