From 794db3702b0bb5f74aa94b09b22a27ac138eedf2 Mon Sep 17 00:00:00 2001 From: Wuwei Lin Date: Tue, 29 Oct 2019 08:23:23 -0400 Subject: [PATCH] Update have_int8 condition to run on compute capability 7.x devices (#4214) --- python/tvm/contrib/nvcc.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/tvm/contrib/nvcc.py b/python/tvm/contrib/nvcc.py index 37e4f2a..05ee338 100644 --- 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 -- 2.7.4