from setuptools import setup
from torch.utils.cpp_extension import CppExtension, CUDAExtension, BuildExtension
import os

this_dir = os.path.dirname(os.path.realpath(__file__))

include_dirs = [
]

nvcc_args = [
]

setup(
  name='ext',
  ext_modules=[
    CppExtension('ext_cpu', ['ext/ext_cpu.cpp']),
    CUDAExtension('ext_cuda', ['ext/ext_cuda.cpp', 'ext/ext_kernel.cu'], extra_compile_args={'cxx': [], 'nvcc': nvcc_args}),
  ],
  cmdclass={'build_ext': BuildExtension},
  include_dirs=include_dirs
)

b1a43fc65b92e3ba8763a5018d7141cef1c9769a
<mask token>

<mask token>
setup(name='ext', ext_modules=[CppExtension('ext_cpu', ['ext/ext_cpu.cpp']),
    CUDAExtension('ext_cuda', ['ext/ext_cuda.cpp', 'ext/ext_kernel.cu'],
    extra_compile_args={'cxx': [], 'nvcc': nvcc_args})], cmdclass={
    'build_ext': BuildExtension}, include_dirs=include_dirs)

<mask token>
this_dir = os.path.dirname(os.path.realpath(__file__))
include_dirs = []
nvcc_args = []
setup(name='ext', ext_modules=[CppExtension('ext_cpu', ['ext/ext_cpu.cpp']),
    CUDAExtension('ext_cuda', ['ext/ext_cuda.cpp', 'ext/ext_kernel.cu'],
    extra_compile_args={'cxx': [], 'nvcc': nvcc_args})], cmdclass={
    'build_ext': BuildExtension}, include_dirs=include_dirs)

from setuptools import setup
from torch.utils.cpp_extension import CppExtension, CUDAExtension, BuildExtension
import os
this_dir = os.path.dirname(os.path.realpath(__file__))
include_dirs = []
nvcc_args = []
setup(name='ext', ext_modules=[CppExtension('ext_cpu', ['ext/ext_cpu.cpp']),
    CUDAExtension('ext_cuda', ['ext/ext_cuda.cpp', 'ext/ext_kernel.cu'],
    extra_compile_args={'cxx': [], 'nvcc': nvcc_args})], cmdclass={
    'build_ext': BuildExtension}, include_dirs=include_dirs)

from setuptools import setup
from torch.utils.cpp_extension import CppExtension, CUDAExtension, BuildExtension
import os

this_dir = os.path.dirname(os.path.realpath(__file__))

include_dirs = [
]

nvcc_args = [
]

setup(
  name='ext',
  ext_modules=[
    CppExtension('ext_cpu', ['ext/ext_cpu.cpp']),
    CUDAExtension('ext_cuda', ['ext/ext_cuda.cpp', 'ext/ext_kernel.cu'], extra_compile_args={'cxx': [], 'nvcc': nvcc_args}),
  ],
  cmdclass={'build_ext': BuildExtension},
  include_dirs=include_dirs
)
