mirror of
https://github.com/tmux-plugins/tmux-cpu.git
synced 2026-09-10 07:16:16 -04:00
28 lines
749 B
Bash
Executable file
28 lines
749 B
Bash
Executable file
#!/usr/bin/env bash
|
|
|
|
CURRENT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
|
|
# shellcheck source=scripts/helpers.sh
|
|
source "$CURRENT_DIR/helpers.sh"
|
|
|
|
gram_total_format="%dk"
|
|
|
|
print_gram_total() {
|
|
gram_total_format=$(get_tmux_option "@gram_total_format" "$gram_total_format")
|
|
|
|
if command_exists "nvidia-smi"; then
|
|
loads=$(cached_eval nvidia-smi | sed -nr 's/.*\s([0-9]+)MiB\s*\/\s*([0-9]+)MiB.*/\1 \2/p')
|
|
elif command_exists "cuda-smi"; then
|
|
loads=$(cached_eval cuda-smi | sed -nr 's/.*\s([0-9.]+) of ([0-9.]+) MB.*/\1 \2/p' | awk '{print $2-$1" "$2}')
|
|
else
|
|
echo "No GPU"
|
|
return
|
|
fi
|
|
echo "$loads" | awk -v format="$gram_total_format" '{total+=$2} END {printf format, total}'
|
|
}
|
|
|
|
main() {
|
|
print_gram_total
|
|
}
|
|
main "$@"
|