Mirrored_Repos/nvidia-patch
1
0
Fork 0
mirror of https://github.com/keylase/nvidia-patch.git synced 2024-11-26 15:37:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

linux: patch.sh: new opmode - query for specific version support

Browse Source
This commit is contained in:
Vladislav Yarmak 2019-09-27 20:00:31 +03:00
parent f25b0fd056
commit 1029bfb2a1
1 changed files with 26 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
patch.sh
View File

@ -9,14 +9,16 @@ silent_flag=''
print_usage() { printf ' print_usage() { printf '
SYNOPSIS SYNOPSIS
patch.sh [-s] [-r|-h] patch.sh [-s] [-r|-h|-c VERSION]
DESCRIPTION DESCRIPTION
The patch for Nvidia drivers to increase encoder sessions The patch for Nvidia drivers to remove NVENC session limit
-s Silent mode (No output) -s Silent mode (No output)
-r Rollback to original (Restore lib from backup) -r Rollback to original (Restore lib from backup)
-h Print this help message -h Print this help message
-c VERSION Check if version VERSION supported by this patch.
Returns true exit code (0) if version is supported.
' '
} }
@ -24,11 +26,12 @@ DESCRIPTION
# shellcheck disable=SC2209 # shellcheck disable=SC2209
opmode="patch" opmode="patch"
while getopts 'rsh' flag; do while getopts 'rshc:' flag; do
case "${flag}" in case "${flag}" in
r) opmode="${opmode}rollback" ;; r) opmode="${opmode}rollback" ;;
s) silent_flag='true' ;; s) silent_flag='true' ;;
h) opmode="${opmode}help" ;; h) opmode="${opmode}help" ;;
c) opmode="${opmode}checkversion" ; checked_version="$OPTARG" ;;
*) echo "Incorrect option specified in command line" ; exit 2 ;; *) echo "Incorrect option specified in command line" ; exit 2 ;;
esac esac
done done
@ -107,6 +110,11 @@ declare -A object_list=(
["435.21"]='libnvcuvid.so' ["435.21"]='libnvcuvid.so'
) )
check_version_supported () {
local ver="$1"
[[ "${patch_list[$ver]+isset}" && "${object_list[$ver]+isset}" ]]
}
patch_common () { patch_common () {
NVIDIA_SMI="$(command -v nvidia-smi || true)" NVIDIA_SMI="$(command -v nvidia-smi || true)"
if [[ ! "$NVIDIA_SMI" ]] ; then if [[ ! "$NVIDIA_SMI" ]] ; then
@ -121,7 +129,7 @@ patch_common () {
echo "Detected nvidia driver version: $driver_version" echo "Detected nvidia driver version: $driver_version"
if [[ ! "${patch_list[$driver_version]+isset}" || ! "${object_list[$driver_version]+isset}" ]]; then if ! check_version_supported "$driver_version" ; then
echo "Patch for this ($driver_version) nvidia driver not found." 1>&2 echo "Patch for this ($driver_version) nvidia driver not found." 1>&2
echo "Available patches for: " 1>&2 echo "Available patches for: " 1>&2
for drv in "${!patch_list[@]}"; do for drv in "${!patch_list[@]}"; do
@ -180,9 +188,20 @@ patch () {
echo "Patched!" echo "Patched!"
} }
query_version_support () {
if check_version_supported "$checked_version" ; then
echo "SUPPORTED"
exit 0
else
echo "NOT SUPPORTED"
exit 1
fi
}
case "${opmode}" in case "${opmode}" in
patch) patch ;; patch) patch ;;
patchrollback) rollback ;; patchrollback) rollback ;;
patchhelp) print_usage ; exit 2 ;; patchhelp) print_usage ; exit 2 ;;
patchcheckversion) query_version_support ;;
*) echo "Incorrect combination of flags"; exit 2 ;; *) echo "Incorrect combination of flags"; exit 2 ;;
esac esac