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

794db370 · Wuwei Lin · masahi · e8899285 · 794db370
Commit 794db370 authored Oct 29, 2019 by Wuwei Lin Committed by masahi Oct 29, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

python/tvm/contrib/nvcc.py
+2 -2

No files found.
--- 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