[arch-commits] Commit in python-pytorch/trunk (PKGBUILD)

[Konstantin Gizdov]

    Date: Saturday, April 13, 2019 @ 17:50:51
  Author: kgizdov
Revision: 451461

move C++ API to correct location

Modified:
  python-pytorch/trunk/PKGBUILD

----------+
 PKGBUILD |   40 +++++++++++++++++++++++++++++++++++++++-
 1 file changed, 39 insertions(+), 1 deletion(-)

Modified: PKGBUILD
===================================================================
--- PKGBUILD	2019-04-13 17:35:05 UTC (rev 451460)
+++ PKGBUILD	2019-04-13 17:50:51 UTC (rev 451461)
@@ -5,7 +5,7 @@
 pkgname=("python-pytorch" "python-pytorch-cuda")
 _pkgname="pytorch"
 pkgver=1.0.1
-pkgrel=6
+pkgrel=7
 pkgdesc="Tensors and Dynamic neural networks in Python with strong GPU acceleration"
 arch=('x86_64')
 url="https://pytorch.org"
@@ -15,6 +15,10 @@
 source=("${_pkgname}-${pkgver}::git+https://github.com/pytorch/pytorch.git#tag=v$pkgver")
 sha256sums=('SKIP')
 
+get_pyver () {
+    python -c 'import sys; print(str(sys.version_info[0]) + "." + str(sys.version_info[1]))'
+}
+
 prepare() {
   cd "${_pkgname}-${pkgver}"
 
@@ -63,6 +67,23 @@
   cd "$srcdir/${_pkgname}-${pkgver}"
   python setup.py install --root="$pkgdir"/ --optimize=1 --skip-build
   install -Dm644 LICENSE "${pkgdir}/usr/share/licenses/${pkgname}/LICENSE.txt"
+  # put CMake files in correct place
+  install -d "${pkgdir}/usr/lib/cmake"
+  mv "${pkgdir}/usr/lib/python$(get_pyver)/site-packages/torch/share/cmake"/* \
+     "${pkgdir}/usr/lib/cmake/"
+  # put C++ API in correct place
+  install -d "${pkgdir}/usr/bin"
+  install -d "${pkgdir}/usr/include"
+  install -d "${pkgdir}/usr/lib"
+  mv "${pkgdir}/usr/lib/python$(get_pyver)/site-packages/torch/lib/include"/* \
+     "${pkgdir}/usr/include/"
+  mv "${pkgdir}/usr/lib/python$(get_pyver)/site-packages/torch/lib/THCUNN.h" \
+     "${pkgdir}/usr/lib/python$(get_pyver)/site-packages/torch/lib/THNN.h" \
+     "${pkgdir}/usr/include/"
+  mv "${pkgdir}/usr/lib/python$(get_pyver)/site-packages/torch/lib"/*.so \
+     "${pkgdir}/usr/lib/"
+  mv "${pkgdir}/usr/lib/python$(get_pyver)/site-packages/torch/lib"/torch_shm_manager \
+     "${pkgdir}/usr/bin/"
 }
 
 package_python-pytorch-cuda() {
@@ -72,6 +93,23 @@
   cd "$srcdir/${_pkgname}-${pkgver}-cuda"
   python setup.py install --root="$pkgdir"/ --optimize=1 --skip-build
   install -Dm644 LICENSE "${pkgdir}/usr/share/licenses/${pkgname}/LICENSE.txt"
+  # put CMake files in correct place
+  install -d "${pkgdir}/usr/lib/cmake"
+  mv "${pkgdir}/usr/lib/python$(get_pyver)/site-packages/torch/share/cmake"/* \
+     "${pkgdir}/usr/lib/cmake/"
+  # put C++ API in correct place
+  install -d "${pkgdir}/usr/bin"
+  install -d "${pkgdir}/usr/include"
+  install -d "${pkgdir}/usr/lib"
+  mv "${pkgdir}/usr/lib/python$(get_pyver)/site-packages/torch/lib/include"/* \
+     "${pkgdir}/usr/include/"
+  mv "${pkgdir}/usr/lib/python$(get_pyver)/site-packages/torch/lib/THCUNN.h" \
+     "${pkgdir}/usr/lib/python$(get_pyver)/site-packages/torch/lib/THNN.h" \
+     "${pkgdir}/usr/include/"
+  mv "${pkgdir}/usr/lib/python$(get_pyver)/site-packages/torch/lib"/*.so \
+     "${pkgdir}/usr/lib/"
+  mv "${pkgdir}/usr/lib/python$(get_pyver)/site-packages/torch/lib"/torch_shm_manager \
+     "${pkgdir}/usr/bin/"
 }
 
 # vim:set ts=2 sw=2 et: