Spaces:
Sleeping
Sleeping
File size: 1,043 Bytes
79df973 |
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 |
# -*- coding: utf-8 -*-
#
# @File: setup.py
# @Author: Jiaxiang Tang (@ashawkey)
# @Date: 2023-04-15 10:33:32
# @Last Modified by: Haozhe Xie
# @Last Modified at: 2023-04-29 10:47:10
# @Email: ashawkey1999@gmail.com
# @Ref: https://github.com/ashawkey/torch-ngp
from setuptools import setup
from torch.utils.cpp_extension import BuildExtension, CUDAExtension
setup(
name="grid_encoder",
version="1.0.0",
ext_modules=[
CUDAExtension(
name="grid_encoder_ext",
sources=[
"grid_encoder_ext.cu",
"bindings.cpp",
],
extra_compile_args={
"cxx": ["-O3", "-std=c++14"],
"nvcc": [
"-O3",
"-std=c++14",
"-U__CUDA_NO_HALF_OPERATORS__",
"-U__CUDA_NO_HALF_CONVERSIONS__",
"-U__CUDA_NO_HALF2_OPERATORS__",
],
},
),
],
cmdclass={
"build_ext": BuildExtension,
},
)
|