from setuptools import setup, Extension
from torch.utils import cpp_extension

setup(name='torch_stablesort',
    ext_modules=[cpp_extension.CppExtension('torch_stablesort_cpp',
        ['torch_stablesort.cpp'])],
            cmdclass={'build_ext': cpp_extension.BuildExtension})