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

Bring back C++ extension torch.h (#7310)

Browse Source 
* Bring back C++ extension torch.h

* Fix python.h include in python_tensor.cpp
This commit is contained in:
Peter Goldsborough
2018-05-05 14:06:27 -07:00
committed by GitHub
parent 6087a5feaa
commit 54a4867675
11 changed files with 14 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/cpp_extensions/cuda_extension.cpp
View File

@ -1,4 +1,4 @@
#include <torch/python.h>
#include <torch/torch.h>
// Declare the function from cuda_extension.cu. It will be compiled
// separately with nvcc and linked with the object file of cuda_extension.cpp