Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
setup.py 234 Bytes
Newer Older
rusty1s's avatar
first aten try  
rusty1s committed
1
2
3
4
5
6
7
8 
from setuptools import setup
from torch.utils.cpp_extension import BuildExtension, CppExtension

setup(
    name='cluster',
    ext_modules=[CppExtension('cluster_cpu', ['cluster.cpp'])],
    cmdclass={'build_ext': BuildExtension},
)