Creates a "setuptools.Extension" for CUDA/C++.
Convenience method that creates a "setuptools.Extension" with the
   bare minimum (but often sufficient) arguments to build a CUDA/C++
   extension. This includes the CUDA include path, library path and
   runtime library.
All arguments are forwarded to the "setuptools.Extension"
   constructor.
-[ Example ]-



from setuptools import setup
from torch.utils.cpp_extension import BuildExtension, CUDAExtension
setup(
   ...     name='cuda_extension',
   ...     ext_modules=[
   ...         CUDAExtension(
   ...                 name='cuda_extension',
   ...                 sources=['extension.cpp', 'extension_kernel.cu'],
   ...                 extra_compile_args={'cxx': ['-g'],
   ...                                     'nvcc': ['-O2']})
   ...     ],
   ...     cmdclass={
   ...         'build_ext': BuildExtension
   ...     })



Compute capabilities:
By default the extension will be compiled to run on all archs of