From 2c38eca752b08203ac8c576510c9f2e135e3c821 Mon Sep 17 00:00:00 2001
From: Vladislav Yarmak <vladislav@vm-0.com>
Date: Sat, 12 Oct 2019 11:32:00 +0300
Subject: [PATCH] ndl: track tesla drivers as well

---
 tools/nv-driver-locator/get_nvidia_downloads.py       | 2 ++
 tools/nv-driver-locator/nv-driver-locator.json.sample | 9 +++++++++
 2 files changed, 11 insertions(+)

diff --git a/tools/nv-driver-locator/get_nvidia_downloads.py b/tools/nv-driver-locator/get_nvidia_downloads.py
index 0e70a8a..4132d04 100755
--- a/tools/nv-driver-locator/get_nvidia_downloads.py
+++ b/tools/nv-driver-locator/get_nvidia_downloads.py
@@ -56,6 +56,7 @@ class Product(enum.Enum):
     GeForceMobile = (111, 890)
     Quadro = (73, 844)
     QuadroMobile = (74, 875)
+    Tesla = (110, 883)
 
     def __str__(self):
         return self.name
@@ -91,6 +92,7 @@ class DriverLanguage(enum.Enum):
 class CUDAToolkitVersion(enum.Enum):
     Nothing = 0
     v10_0 = 20
+    v10_1 = 21
 
     def __str__(self):
         return self.name
diff --git a/tools/nv-driver-locator/nv-driver-locator.json.sample b/tools/nv-driver-locator/nv-driver-locator.json.sample
index 7bfb820..4325f35 100644
--- a/tools/nv-driver-locator/nv-driver-locator.json.sample
+++ b/tools/nv-driver-locator/nv-driver-locator.json.sample
@@ -225,6 +225,15 @@
                 "certlevel": "Certified"
             }
         },
+        {
+            "type": "nvidia_downloads",
+            "name": "downloads linux tesla all",
+            "params": {
+                "os": "Linux_64",
+                "product": "Tesla",
+                "certlevel": "All"
+            }
+        },
         {
             "type": "cuda_downloads",
             "name": "cuda toolkit tracker",