pkgbuild_cpu.diff

diff --git a/repos/community-x86_64/PKGBUILD b/repos/community-x86_64/PKGBUILD
index 43ffc28..f9537c6 100644
--- a/repos/community-x86_64/PKGBUILD
+++ b/repos/community-x86_64/PKGBUILD
@@ -2,7 +2,7 @@
 # Contributor: Stephen Zhang <zsrkmyn at gmail dot com>
 
 pkgbase=python-pytorch
-pkgname=("python-pytorch" "python-pytorch-cuda")
+pkgname=("python-pytorch")
 _pkgname="pytorch"
 pkgver=1.10.2
 _pkgver=1.10.2
@@ -12,11 +12,11 @@ pkgdesc="${_pkgdesc}"
 arch=('x86_64')
 url="https://pytorch.org"
 license=('BSD')
-depends=('google-glog' 'gflags' 'opencv' 'openmp' 'nccl' 'pybind11' 'python' 'python-yaml' 'libuv'
+depends=('google-glog' 'gflags' 'opencv' 'openmp' 'pybind11' 'python' 'python-yaml' 'libuv'
          'python-numpy' 'protobuf' 'ffmpeg' 'python-future' 'qt5-base' 'onednn' 'intel-mkl'
          'python-typing_extensions')
-makedepends=('python' 'python-setuptools' 'python-yaml' 'python-numpy' 'cmake' 'cuda'
-             'cudnn' 'git' 'magma' 'ninja' 'pkgconfig' 'doxygen')
+makedepends=('python' 'python-setuptools' 'python-yaml' 'python-numpy' 'cmake'
+             'git' 'ninja' 'pkgconfig' 'doxygen')
 source=("${_pkgname}-${pkgver}::git+https://github.com/pytorch/pytorch.git#tag=v$_pkgver"
         # generated using parse-submodules
         "${pkgname}-ios-cmake::git+https://github.com/Yangqing/ios-cmake.git"
@@ -239,20 +239,20 @@ build() {
   export USE_CUDA=0
   export USE_CUDNN=0
   cd "${srcdir}/${_pkgname}-${pkgver}"
-  echo "add_definitions(-march=haswell)" >> cmake/MiscCheck.cmake
+  # echo "add_definitions(-march=haswell)" >> cmake/MiscCheck.cmake
   # this horrible hack is necessary because the current release
   # ships inconsistent CMake which tries to build objects before
   # thier dependencies, build twice when dependencies are available
   python setup.py build || python setup.py build
 
 
-  echo "Building with cuda and with non-x86-64 optimizations"
-  export USE_CUDA=1
-  export USE_CUDNN=1
-  cd "${srcdir}/${_pkgname}-${pkgver}-cuda"
-  echo "add_definitions(-march=haswell)" >> cmake/MiscCheck.cmake
-  # same horrible hack as above
-  python setup.py build || python setup.py build
+  # echo "Building with cuda and with non-x86-64 optimizations"
+  # export USE_CUDA=1
+  # export USE_CUDNN=1
+  # cd "${srcdir}/${_pkgname}-${pkgver}-cuda"
+  # echo "add_definitions(-march=haswell)" >> cmake/MiscCheck.cmake
+  # # same horrible hack as above
+  # python setup.py build || python setup.py build
 }
 
 _package() {
@@ -293,15 +293,15 @@ package_python-pytorch() {
   _package
 }
 
-package_python-pytorch-cuda() {
-  pkgdesc="${_pkgdesc} (with CUDA and AVX2 CPU optimizations)"
-  depends+=(cuda cudnn magma)
-  replaces=(python-pytorch-opt-cuda)
-  conflicts=(python-pytorch)
-  provides=(python-pytorch)
-
-  cd "${srcdir}/${_pkgname}-${pkgver}-cuda"
-  _package
-}
-
+# package_python-pytorch-cuda() {
+#   pkgdesc="${_pkgdesc} (with CUDA and AVX2 CPU optimizations)"
+#   depends+=(cuda cudnn magma)
+#   replaces=(python-pytorch-opt-cuda)
+#   conflicts=(python-pytorch)
+#   provides=(python-pytorch)
+#
+#   cd "${srcdir}/${_pkgname}-${pkgver}-cuda"
+#   _package
+# }
+#
 # vim:set ts=2 sw=2 et: