Skip to content

Commit ee77399

Browse files
committed
fix vfio PCI id templating
1 parent 8d26471 commit ee77399

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

etc/kayobe/ansible/pci-passthrough.yml

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,9 +9,15 @@
99
ansible_ssh_common_args: "-o StrictHostKeyChecking=no"
1010
ansible_python_interpreter: "/usr/bin/python3"
1111
vfio_pci_ids: |-
12+
{% set gpu_list = [] %}
1213
{% set output = [] %}
13-
{% for gpu_type in gpu_group_map | dict2items | map(attribute='value') | flatten | unique | default([]) %}
14-
{% set _ = output.append(stackhpc_gpu_data[gpu_type]['vendor_id'] + ':' + stackhpc_gpu_data[gpu_type]['product_id']) %}
14+
{% for gpu_type in gpu_group_map | dict2items | default([]) %}
15+
{% if gpu_type.key in group_names %}
16+
{% set _ = gpu_list.append(gpu_type.value) %}
17+
{% endif %}
18+
{% endfor %}
19+
{% for item in gpu_list | flatten | unique %}
20+
{% set _ = output.append(stackhpc_gpu_data[item]['vendor_id'] + ':' + stackhpc_gpu_data[item]['product_id']) %}
1521
{% endfor %}
1622
{{ output | join(',') }}
1723
reboot_timeout_s: "{{ 20 * 60 }}"

0 commit comments

Comments
 (0)