1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
#!/bin/bash
# autopkgtest podman backend with support for AMD GPUs in rootless containers
#
# This is just a thin wrapper around autopkgtest-virt-podman. It adds all the
# arguments necessary for using AMD GPUs in the container.
#
# Author: Christian Kastner <ckk@kvr.at>
# License: MIT
function usage() {
cat >&2 <<- EOF
autopkgtest podman backend with support for AMD GPUs
This is a simple wrapper around autopkgtest-virt-podman that sets up the
correct options to pass in devices and map necessary GIDs. All options on
the command line are passed on directly to autopkgtest-virt-podman, so the
reader is referred to its man page autopkgtest-virt-podman(1).
Synopsis:
$0 -h
$0 autopkgtest [...] -- podman+rocm [options] image [-- extra podman-run args...]
Examples:
# Create an image first, if needed
\$ rocm-podman-create -m http://10.1.2.3:9999/debian rocm/debian:unstable
# Run autopkgtests for src:rocrand, using packages from the Archive
\$ autopkgtest -B rocrand -- podman+rocm rocm/debian:unstable
EOF
exit 0
}
[ "${1:-}" = "-h" ] && usage
userNAME=$(whoami)
renderGID="$(getent group render | cut -d: -f3)"
# By policy
videoGID=44
# Sanity checks
if [ -z "$renderGID" ]
then
cat >&2 <<-EOF
Group 'render' does not exist on this system. Are you sure that you are on
the right system? This group should have been autmatically created by the
udev package."
EOF
exit 1
elif ! groups "$userNAME" | grep -q '\brender\b'
then
echo "'$userNAME' is not in group 'render'." >&2
exit 1
elif ! groups "$userNAME" | grep -q '\bvideo\b'
then
echo "'$userNAME' is not in group 'video'." >&2
exit 1
elif [ "$(cat /proc/sys/kernel/unprivileged_userns_clone)" != "1" ]
then
echo "unprivileged_userns_clone not enabled.">&2
exit 1
elif ! [ -c /dev/kfd ]
then
echo "Device /dev/kfd does not exist - is the amdgpu module loaded?." >&2
exit 1
elif ! [ -w /dev/kfd ]
then
echo "No write permissions for /dev/kfd." >&2
exit 1
elif ! grep -q "$userNAME:$renderGID:1" /etc/subgid
then
echo "No subgid mapping for group 'render'. Run rocm-podman-setup" >&2
exit 1
elif ! grep -q "$userNAME:$videoGID:1" /etc/subgid
then
echo "No subgid mapping for group 'video'. Run rocm-podman-setup" >&2
exit 1
elif ! grep -q -E "$userNAME:[0-9]{6,}:6553[4-6]" /etc/subgid
then
echo "No large subgid mapping for '$(whoami)'. Run rocm-podman-setup" >&2
exit 1
fi
exec autopkgtest-virt-podman "$@" \
--device=/dev/dri \
--device=/dev/kfd \
--gidmap=0:0:1 \
--gidmap=44:1:1 \
--gidmap="$renderGID":2:1 \
--gidmap=1:3:43 \
--gidmap=45:46:$(("$renderGID"-"$videoGID"-1)) \
--gidmap=$(("$renderGID"+1)):$(("$renderGID"+2)):65429
|
