cudnn-tweaks.patch

diff --git a/PKGBUILD b/PKGBUILD
index 0052c3a..ed48961 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -10,31 +10,30 @@ depends=('cuda')
 source=()
 sha256sums=()
 
-_cudnnSrcDir=""
+_cudnnSrcDir="$(realpath .)"
 _cudnnSrcName="cudnn-7.5-linux-x64-v5.0-ga.tgz"
 _cudnnSha256="c4739a00608c3b66a004a74fc8e721848f9112c5cb15f730c1be4964b3a23b3a"
 
 prepare() {
+
   echo "###################################################################"
   echo "# cuDNN is only available to registered CUDA developers.          #"
   echo "# Register on the website, and download the cuDNN source tarball. #"
   echo "###################################################################"
   echo
-  echo -n "Enter the directory containing the cuDNN source tarball ${_cudnnSrcName}: "
-  read _cudnnSrcDir
 
-  # Check for file existence
-  if [[ -f "${_cudnnSrcDir}/${_cudnnSrcName}" ]]; then
-    # Check for file validity
-    sha256sum -c <(printf "${_cudnnSha256}  ${_cudnnSrcDir}/${_cudnnSrcName}\n") || return 1
+  while [ ! -f "${_cudnnSrcDir}/${_cudnnSrcName}" ]; do
+    echo "Error: ${_cudnnSrcName} not found in $(realpath ${_cudnnSrcDir})"
+    echo -n "Enter the directory containing the cuDNN source tarball ${_cudnnSrcName}: "
+    read _cudnnSrcDir
+  done
 
-    # Untar
-    mkdir -p "${srcdir}/${pkgname}-${pkgver}"
-    tar -xzvf "${_cudnnSrcDir}/${_cudnnSrcName}" -C "${srcdir}/${pkgname}-${pkgver}"
-  else
-    echo "Error: ${_cudnnSrcName} not found in ${_cudnnSrcDir}"
-    return 1
-  fi
+  # Check for file validity
+  sha256sum -c <(printf "${_cudnnSha256}  ${_cudnnSrcDir}/${_cudnnSrcName}\n") || return 1
+
+  # Untar
+  mkdir -p "${srcdir}/${pkgname}-${pkgver}"
+  tar -xzvf "${_cudnnSrcDir}/${_cudnnSrcName}" -C "${srcdir}/${pkgname}-${pkgver}"
 }
 
 package() {