<filename>mmdet/ops/roi_align/setup.py<gh_stars>10-100
from setuptools import setup
from torch.utils.cpp_extension import BuildExtension, CUDAExtension
import os
# os.environ['CC']= '/usr/local/gcc-5.3.0/bin/g++'
# os.environ['CXX']= '/usr/local/gcc-5.3.0/bin/g++'
# CC = os.environ.get("CC", None)
# CC_arg = "-ccbin={}".format(CC)

setup(
    name='roi_align_cuda',
    ext_modules=[
        CUDAExtension('roi_align_cuda', [
            'src/roi_align_cuda.cpp',
            'src/roi_align_kernel.cu',
        ],
        extra_compile_args={
            'cxx': [],
            'nvcc': [
                '-D__CUDA_NO_HALF_OPERATORS__',
                '-D__CUDA_NO_HALF_CONVERSIONS__',
                '-D__CUDA_NO_HALF2_OPERATORS__',
                # CC_arg,
                
            ]
        }),
    ],
    cmdclass={'build_ext': BuildExtension})

<filename>mmdet/ops/roi_align/setup.py<gh_stars>10-100
from setuptools import setup
from torch.utils.cpp_extension import BuildExtension, CUDAExtension
import os
# os.environ['CC']= '/usr/local/gcc-5.3.0/bin/g++'
# os.environ['CXX']= '/usr/local/gcc-5.3.0/bin/g++'
# CC = os.environ.get("CC", None)
# CC_arg = "-ccbin={}".format(CC)

setup(
    name='roi_align_cuda',
    ext_modules=[
        CUDAExtension('roi_align_cuda', [
            'src/roi_align_cuda.cpp',
            'src/roi_align_kernel.cu',
        ],
        extra_compile_args={
            'cxx': [],
            'nvcc': [
                '-D__CUDA_NO_HALF_OPERATORS__',
                '-D__CUDA_NO_HALF_CONVERSIONS__',
                '-D__CUDA_NO_HALF2_OPERATORS__',
                # CC_arg,
                
            ]
        }),
    ],
    cmdclass={'build_ext': BuildExtension})

# os.environ['CC']= '/usr/local/gcc-5.3.0/bin/g++' # os.environ['CXX']= '/usr/local/gcc-5.3.0/bin/g++' # CC = os.environ.get("CC", None) # CC_arg = "-ccbin={}".format(CC) # CC_arg,