andr3/gotch
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

updated Travis CI

Browse Source
This commit is contained in:
sugarme 2020-10-12 21:21:37 +11:00
parent 808acba11f
commit 5a4a65e35c
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
libtch/lib.go.cpu
View File

@ -5,4 +5,7 @@ package libtch
// #cgo CFLAGS: -D_GLIBCXX_USE_CXX11_ABI=1
// #cgo LDFLAGS: -lstdc++ -ltorch -lc10 -ltorch_cpu -L/lib64
// #cgo CXXFLAGS: -std=c++17 -I${SRCDIR} -g -O3
// #cgo CFLAGS: -I${SRCDIR}/libtorch/include -I${SRCDIR}/libtorch/include/torch/csrc/api/include
// #cgo LDFLAGS: -L${SRCDIR}/libtorch/lib
// #cgo CXXFLAGS: -isystem ${SRCDIR}/libtorch/lib ${SRCDIR}/libtorch/include ${SRCDIR}/libtorch/include/torch/csrc/api/include ${SRCDIR}/libtorch/include/torch/csrc
import "C"