IF YOU WOULD LIKE TO GET AN ACCOUNT, please write an email to s dot adaszewski at gmail dot com. User accounts are meant only to report issues and/or generate pull requests. This is a purpose-specific Git hosting for ADARED projects. Thank you for your understanding!
sadaszewski1
/
decagon-pytorch
forked from sadaszewski/decagon-pytorch
1
0
Fork 0
Code Issues 0 Pull Requests 0 Releases 0 Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
224 Commits
1 Branch
866KB
Tree: 747156a971
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '747156a971'
${ noResults }
decagon-pytorch/src/torch_stablesort/setup.py

9 lines
338B
Raw Blame History 

  1. from setuptools import setup, Extension

  2. from torch.utils import cpp_extension

  3. 

  4. setup(name='torch_stablesort',

  5.     ext_modules=[cpp_extension.CppExtension('torch_stablesort_cpp',

  6.         ['torch_stablesort.cpp'],

  7.         extra_compile_args=['-fopenmp', '-ggdb'])],

  8.             cmdclass={'build_ext': cpp_extension.BuildExtension})