Update have_int8 condition to run on compute capability 7.x devices (#…

…4214)
apache · Oct 29, 2019 · 794db37 · 794db37
1 parent e889928
commit 794db37
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/tvm/contrib/nvcc.py b/python/tvm/contrib/nvcc.py
@@ -245,8 +245,8 @@ def have_int8(compute_version):
     compute_version : str
         compute capability of a GPU (e.g. "6.1")
     """
-    major, minor = parse_compute_version(compute_version)
-    if major == 6 and minor == 1:
+    major, _ = parse_compute_version(compute_version)
+    if major >= 6:
         return True
 
     return False