Fix device detection regex for NVIDIA GPUs. Missed GM20x (Maxwell) like Tesla M40

Author: cmdr2

tests/test_platform_detection.py

@@ -254,6 +254,17 @@ def test_multiple_gpu_NVIDIA(monkeypatch):
     assert get_torch_platform(gpu_infos) == expected
 
 
+def test_multiple_gpu_NVIDIA_maxwell(monkeypatch):
+    monkeypatch.setattr("torchruntime.platform_detection.os_name", "Linux")
+    monkeypatch.setattr("torchruntime.platform_detection.arch", "x86_64")
+    gpu_infos = [
+        GPU(NVIDIA, "NVIDIA", "17fd", "GM200GL [Tesla M40]", True),
+        GPU(NVIDIA, "NVIDIA", "1401", "GM206 [GeForce GTX 960]", True),
+    ]
+    expected = "cu124"
+    assert get_torch_platform(gpu_infos) == expected
+
+
 def test_multiple_gpu_AMD_Navi3_Navi2(monkeypatch, capsys):
     monkeypatch.setattr("torchruntime.platform_detection.os_name", "Linux")
     monkeypatch.setattr("torchruntime.platform_detection.arch", "x86_64")

torchruntime/platform_detection.py

@@ -10,13 +10,13 @@
 py_version = sys.version_info
 
 # https://www.techpowerup.com/gpu-specs/?architecture=Kepler&sort=generation and so on (change the arch field)
-KEPLER_DEVICES = re.compile(r"\b(gk1\d{2}\w*)\b", re.IGNORECASE)  # sm3.7
-MAXWELL_DEVICES = re.compile(r"\b(gm10\d\w*)\b", re.IGNORECASE)  # sm5
-PASCAL_DEVICES = re.compile(r"\b(gp10\d\w*)\b", re.IGNORECASE)  # sm6
-VOLTA_DEVICES = re.compile(r"\b(gv100\w*)\b", re.IGNORECASE)  # sm7
-TURING_DEVICES = re.compile(r"\b(tu1\d{2}\w*)\b", re.IGNORECASE)  # sm7.5
-AMPERE_DEVICES = re.compile(r"\b(ga10\d\w*)\b", re.IGNORECASE)  # sm8.6
-ADA_LOVELACE_DEVICES = re.compile(r"\b(ad10\d\w*)\b", re.IGNORECASE)  # sm8.9
+KEPLER_DEVICES = re.compile(r"\b(gk\d+\w*)\b", re.IGNORECASE)  # sm3.7
+MAXWELL_DEVICES = re.compile(r"\b(gm\d+\w*)\b", re.IGNORECASE)  # sm5
+PASCAL_DEVICES = re.compile(r"\b(gp\d+\w*)\b", re.IGNORECASE)  # sm6
+VOLTA_DEVICES = re.compile(r"\b(gv\d+\w*)\b", re.IGNORECASE)  # sm7
+TURING_DEVICES = re.compile(r"\b(tu\d+\w*)\b", re.IGNORECASE)  # sm7.5
+AMPERE_DEVICES = re.compile(r"\b(ga\d+\w*)\b", re.IGNORECASE)  # sm8.6
+ADA_LOVELACE_DEVICES = re.compile(r"\b(ad\d+\w*)\b", re.IGNORECASE)  # sm8.9
 BLACKWELL_DEVICES = re.compile(r"\b(?:5060|5070|5080|5090)\b", re.IGNORECASE)  # sm10, sm12
 
 NVIDIA_ARCH_MAP = {