Skip to content

Commit

Permalink
Revert
Browse files Browse the repository at this point in the history
  • Loading branch information
xuzhao9 committed Jan 28, 2025
1 parent c981cce commit b196360
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions userbenchmark/release-test/monitor_proc.sh
Original file line number Diff line number Diff line change
Expand Up @@ -16,9 +16,7 @@ get_gpu_max_memory_usage_cuda() {
local my_pid=$1
local max=$2
local curr
# pick the process that uses the most GPU memory
nvidia-smi dmon -s m -c 1 -o T -i 0 | tee mem-log.txt
curr=$(cat mem-log.txt | tail -n +3 | awk '{print $3}' | sort -n | tail -1 | grep -o "[0-9.]*")
curr=$(nvidia-smi dmon -s m -c 1 -o T -i 0 | tail -n +3 | awk '{print $3}' | sort -n | tail -1 | grep -o "[0-9.]*")
# Some processes might not use the GPU, then memory usage should be 0
if [ "${curr}" -eq 0 ] ; then
echo "${max}"
Expand Down

0 comments on commit b196360

Please sign in to comment.