clean up

1 files changed, 6 insertions, 4 deletions

diff --git a/PKGBUILD b/PKGBUILD
index 415deacece15..27010311c97d 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -10,16 +10,18 @@ license=('BSD')
 depends=('cuda' 'cuda-tools' 'onednn' 'cudnn' 'cmake')
 provides=()
 conflicts=()
-source=("libtorch-cuda.zip::https://download.pytorch.org/libtorch/cu121/libtorch-cxx11-abi-shared-with-deps-2.1.0%2Bcu121.zip")
+source=("${pkgname}-${pkgver}.zip::https://download.pytorch.org/libtorch/cu121/libtorch-cxx11-abi-shared-with-deps-2.1.0%2Bcu121.zip")
 sha512sums=('51d6558149a63eb05c2cc4363f1eb0222e81b14003d091e604235eceb4927f85e346ed61ab9d738ddad26b470d95c73efcdcc4335240f28cd418c7ac0789f9e4')
 
 package() {
+  rm "${pkgname}-${pkgver}"
   install -d "${pkgdir}/usr/lib/"
   rm ${srcdir}/libtorch/lib/libcudnn*
-  cp -r ${srcdir}/libtorch/lib/* ${pkgdir}/usr/lib/
+  mv ${srcdir}/libtorch/lib/* ${pkgdir}/usr/lib/
   install -d ${pkgdir}/usr/include/
   rm ${srcdir}/libtorch/include/dnnl*
-  cp -r ${srcdir}/libtorch/include/* ${pkgdir}/usr/include/
+  mv ${srcdir}/libtorch/include/* ${pkgdir}/usr/include/
   install -d ${pkgdir}/usr/share/cmake/
-  cp -r ${srcdir}/libtorch/share/cmake/* ${pkgdir}/usr/share/cmake/
+  mv ${srcdir}/libtorch/share/cmake/* ${pkgdir}/usr/share/cmake/
+  rm -r ${srcdir}
 }