代码拉取完成,页面将自动刷新
from setuptools import setup
import os.path as osp
from torch.utils.cpp_extension import BuildExtension, CUDAExtension
ROOT_DIR = osp.dirname(osp.abspath(__file__))
__version__ = None
exec(open('svox/version.py', 'r').read())
CUDA_FLAGS = []
INSTALL_REQUIREMENTS = []
try:
ext_modules = [
CUDAExtension('svox.csrc', [
'svox/csrc/svox.cpp',
'svox/csrc/svox_kernel.cu',
'svox/csrc/rt_kernel.cu',
'svox/csrc/quantizer.cpp',
], include_dirs=[osp.join(ROOT_DIR, "svox", "csrc", "include")],
optional=True),
]
except:
import warnings
warnings.warn("Failed to build CUDA extension")
ext_modules = []
setup(
name='svox',
version=__version__,
author='Alex Yu',
author_email='alexyu99126@gmail.com',
description='Sparse voxel N^3-tree data structure using CUDA',
long_description='Sparse voxel N^3-tree data structure PyTorch extension, using CUDA',
ext_modules=ext_modules,
setup_requires=['pybind11>=2.5.0'],
packages=['svox', 'svox.csrc'],
cmdclass={'build_ext': BuildExtension},
zip_safe=False,
)
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。