Add a vGPU name check script to make using KVM hypervisors easier (#28)

This commit is contained in:
FIFARenderZ 2021-04-13 20:28:43 +02:00 committed by GitHub
parent c1cdb5ac02
commit 9416363206
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 68 additions and 0 deletions

68
scripts/vgpu-name.sh Normal file
View File

@ -0,0 +1,68 @@
#!/bin/bash
# This file is part of the "vgpu_unlock" project, and is distributed under the MIT License.
# See the LICENSE file for more details.
help_function()
{
echo "This script checks for the internal names of vGPU profiles."
echo
echo "Usage: $0 [-h] [-p <[VGPU_NAME]|ALL>]"
echo "Options:"
echo "-h Display the help page."
echo "-p Display selected VGPU_NAME or ALL profiles."
exit 0;
}
process_id()
{
echo "$(cat "$1"/name): $(basename "$1" | cut -d '/' -f 8)"
}
list_all_ids()
{
for gpu in $(lspci -Dd 10de: -s 0 -n | awk '$2 ~ /030[02]/{print $1}'); do
if [ -d "/sys/bus/pci/devices/$gpu/mdev_supported_types/" ]; then
for gpu_type in /sys/bus/pci/devices/"$gpu"/mdev_supported_types/* ; do
process_id "$gpu_type"
done
exit
fi
done
}
get_pciid()
{
for gpu in $(lspci -Dd 10de: -s 0 -n | awk '$2 ~ /030[02]/{print $1}'); do
if [ -d "/sys/bus/pci/devices/$gpu/mdev_supported_types/" ]; then
if [ ! -z "$(grep -l "$1" /sys/bus/pci/devices/"$gpu"/mdev_supported_types/nvidia-*/name)" ]; then
grep -l "$1" /sys/bus/pci/devices/"$gpu"/mdev_supported_types/nvidia-*/name | cut -d '/' -f 8
exit
fi
fi
done
}
while getopts ":p:" option; do
case "${option}" in
p)
p=${OPTARG}
;;
*)
help_function
;;
esac
done
shift $((OPTIND-1))
if [ -z "${p}" ]; then
help_function
fi
if [ "$p" = "ALL" ]; then
list_all_ids
else
get_pciid "$p"
fi
exit 1