tmux/tmux-plugins.tmux-cpu
1
0
Fork 0
mirror of https://github.com/tmux-plugins/tmux-cpu.git synced 2026-09-10 07:16:16 -04:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge 564de038cf into bcb110d754

Browse source
This commit is contained in:
Toby D 2024-10-06 20:54:50 +01:00 committed by GitHub
parent bcb110d754 564de038cf
commit 2bfb20b792
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 58 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
cpu.tmux
View file

@ -18,6 +18,8 @@ cpu_interpolation=(
"\#{ram_bg_color}"
"\#{ram_fg_color}"
"\#{gram_percentage}"
"\#{gram_total}"
"\#{gram_used}"
"\#{gram_icon}"
"\#{gram_bg_color}"
"\#{gram_fg_color}"
@ -44,6 +46,8 @@ cpu_commands=(
"#($CURRENT_DIR/scripts/ram_bg_color.sh)"
"#($CURRENT_DIR/scripts/ram_fg_color.sh)"
"#($CURRENT_DIR/scripts/gram_percentage.sh)"
"#($CURRENT_DIR/scripts/gram_total.sh)"
"#($CURRENT_DIR/scripts/gram_used.sh)"
"#($CURRENT_DIR/scripts/gram_icon.sh)"
"#($CURRENT_DIR/scripts/gram_bg_color.sh)"
"#($CURRENT_DIR/scripts/gram_fg_color.sh)"

27
scripts/gram_total.sh Executable file
View file

@ -0,0 +1,27 @@
#!/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 "$@"

27
scripts/gram_used.sh Executable file
View file

@ -0,0 +1,27 @@
#!/usr/bin/env bash
CURRENT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
# shellcheck source=scripts/helpers.sh
source "$CURRENT_DIR/helpers.sh"
gram_used_format="%dk"
print_gram_usage() {
gram_used_format=$(get_tmux_option "@gram_used_format" "$gram_used_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_used_format" '{used+=$1} END {printf format, used}'
}
main() {
print_gram_usage
}
main "$@"