Skip to content

Commit

Permalink
Merge branch 'add-gpu-alloc-debug' into next
Browse files Browse the repository at this point in the history
* add-gpu-alloc-debug:
  gpu-setup: Add some temporary debug messages
  • Loading branch information
donald committed Dec 30, 2023
2 parents b7210b6 + e6ed782 commit aa03e45
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions helper/gpu-setup
Original file line number Diff line number Diff line change
Expand Up @@ -235,6 +235,8 @@ job_init() {
pid=$1
uid=$2

echo "XXX $$ job_init $pid: called" >&2

test -d /dev/shm/mxqd/gpu_devs || die "$0: Not initialized (no dir /dev/shm/mxqd/gpu_devs)"

shopt -s nullglob
Expand All @@ -253,6 +255,7 @@ job_init() {
esac
done
cat $d/uuid
echo "XXX $$ job_init $pid: allocated gpu from $d" >&2
exit
fi
done
Expand All @@ -263,11 +266,14 @@ job_release() {
(( $# == 1 )) || usage
pid=$1

echo "XXX $$ job_release $pid: called" >&2

test -d /dev/shm/mxqd/gpu_devs || die "$0: Not initialized (no dir /dev/shm/mxqd/gpu_devs)"
for d in /dev/shm/mxqd/gpu_devs/???; do
if [ -e $d/pid ]; then
test_pid="$(cat $d/pid 2>/dev/null)"
if [ "$pid" = "$test_pid" ]; then
echo "XXX $$ job_release $pid: found my pid in $d, releasing" >&2
rm $d/pid
for f in $(cat $d/access-files); do
case $f in
Expand Down

0 comments on commit aa03e45

Please sign in to comment.