You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
62 lines
1.6 KiB
62 lines
1.6 KiB
#!/usr/bin/python3
|
|
# -*- coding: utf-8 -*-
|
|
#
|
|
# Copyright (C) 2021 Simone Caronni <negativo17@gmail.com>
|
|
# Licensed under the GNU General Public License Version or later
|
|
|
|
import sys
|
|
|
|
def main():
|
|
if len(sys.argv) != 2:
|
|
print("usage: %s README.txt" % sys.argv[0])
|
|
return 1
|
|
|
|
f = open(sys.argv[1])
|
|
in_section = False
|
|
in_table = False
|
|
pids = []
|
|
sections = ["NVIDIA GEFORCE GPUS", "NVIDIA RTX/QUADRO GPUS", "NVIDIA NVS GPUS", "NVIDIA TESLA GPUS", "NVIDIA GRID GPUS"]
|
|
for section in sections:
|
|
|
|
for line in f.readlines():
|
|
|
|
# Find the right data tables
|
|
if line.find(section) != -1:
|
|
in_section = True
|
|
continue
|
|
if not in_section:
|
|
continue
|
|
|
|
# Remove Windows and Linux line endings
|
|
line = line.replace('\r', '')
|
|
line = line.replace('\n', '')
|
|
|
|
# End of section
|
|
if len(line) > 0 and not line.startswith(' '):
|
|
in_section = False
|
|
in_table = False
|
|
continue
|
|
|
|
if len(line) == 0:
|
|
continue
|
|
|
|
# Skip the header
|
|
if line.startswith(' ---'):
|
|
in_table = True
|
|
continue
|
|
if not in_table:
|
|
continue
|
|
|
|
# PCI ID
|
|
pid = int(line[50:54], 16)
|
|
if not pid in pids:
|
|
pids.append(pid)
|
|
|
|
for pid in pids:
|
|
vid = 0x10de
|
|
print("pci:v%08Xd%08Xsv*sd*bc*sc*i*" % (vid, pid))
|
|
|
|
if __name__ == "__main__":
|
|
main()
|
|
|