text: #!/usr/bin/env python3
  import os
  import torch
  
  from setuptools import setup, find_packages
  from torch.utils.cpp_extension import BuildExtension, CUDAExtension
  
  import sys
  sys.path.append("..")
  from compiler_args import nvcc_args, cxx_args
  
  setup(
      name='interpolation_cuda',
      ext_modules=[
          CUDAExtension('interpolation_cuda', [
              'interpolation_cuda.cc',
              'interpolation_cuda_kernel.cu'
          ], extra_compile_args={'cxx': cxx_args, 'nvcc': nvcc_args})
      ],
      cmdclass={
          'build_ext': BuildExtension
      })
  