frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add ffi utils for user C extensions

Browse Source
This commit is contained in:
Adam Paszke
2016-09-28 08:58:53 -07:00
parent 57610a7471
commit 941cf4e63d
13 changed files with 279 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
setup.py
View File

@ -244,6 +244,11 @@ setup(name="torch", version="0.1",
'clean': clean,
},
packages=packages,
package_data={'torch': ['lib/*.so*', 'lib/*.dylib*', 'lib/*.h', 'lib/torch_shm_manager']},
package_data={'torch': [
'lib/*.so*', 'lib/*.dylib*',
'lib/torch_shm_manager',
'lib/*.h',
'lib/include/TH/*.h', 'lib/include/TH/generic/*.h',
'lib/include/THC/*.h', 'lib/include/THC/generic/*.h']},
install_requires=['pyyaml'],
)