Spaces:
Sleeping
Sleeping
File size: 773 Bytes
79df973 1a8fbd0 79df973 1a8fbd0 79df973 1a8fbd0 79df973 1a8fbd0 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 |
# -*- coding: utf-8 -*-
#
# @File: setup.py
# @Author: Haozhe Xie
# @Date: 2023-03-24 20:35:43
# @Last Modified by: Haozhe Xie
# @Last Modified at: 2024-09-22 11:05:34
# @Email: [email protected]
from setuptools import setup
from torch.utils.cpp_extension import BuildExtension, CUDAExtension
cxx_args = []
nvcc_args = ["-O3", "--gpu-architecture=compute_86", "--gpu-code=sm_86,compute_86"]
setup(
name="extrude_tensor",
version="1.0.0",
ext_modules=[
CUDAExtension(
"extrude_tensor_ext",
sources=[
"bindings.cpp",
"extrude_tensor_ext.cu",
],
extra_compile_args={"cxx": cxx_args, "nvcc": nvcc_args},
),
],
cmdclass={"build_ext": BuildExtension},
)
|