device_probe.py: Add Maxwell and Pascal detection

This commit is contained in:
Mykola Grymalyuk
2022-05-07 19:36:02 -06:00
parent b6d89f7310
commit ae983db5f0
3 changed files with 87 additions and 1 deletions

View File

@@ -200,6 +200,8 @@ class NVIDIA(GPU):
Fermi = "Fermi"
Tesla = "Tesla"
Kepler = "Kepler"
Maxwell = "Maxwell"
Pascal = "Pascal"
Unknown = "Unknown"
arch: Archs = field(init=False)
@@ -214,6 +216,10 @@ class NVIDIA(GPU):
self.arch = NVIDIA.Archs.Fermi
elif self.device_id in pci_data.nvidia_ids.kepler_ids:
self.arch = NVIDIA.Archs.Kepler
elif self.device_id in pci_data.nvidia_ids.maxwell_ids:
self.arch = NVIDIA.Archs.Maxwell
elif self.device_id in pci_data.nvidia_ids.pascal_ids:
self.arch = NVIDIA.Archs.Pascal
else:
self.arch = NVIDIA.Archs.Unknown