|
1234567 |
- 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})
|