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

Add exception handling

Browse Source
This commit is contained in:
Adam Paszke
2016-05-05 20:58:13 +02:00
parent fa994baada
commit 842e1b6358
11 changed files with 432 additions and 358 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
setup.py
View File

@ -32,8 +32,9 @@ with open(out_init, 'w') as f:
################################################################################
# WARNING
# This file is generated automatically. Do not edit it, as it will be
# regenerated at next build
# regenerated during next build
################################################################################
""")
f.write(header)
f.write(generated)
@ -41,9 +42,15 @@ with open(out_init, 'w') as f:
################################################################################
# Declare the package
################################################################################
sources = [
"torch/csrc/Module.cpp",
"torch/csrc/Tensor.cpp",
"torch/csrc/Storage.cpp",
]
C = Extension("torch.C",
libraries=['TH'],
sources=["torch/csrc/Module.c", "torch/csrc/Tensor.c", "torch/csrc/Storage.c"],
sources=sources,
language='c++',
include_dirs=["torch/csrc"])