Skip to content

Commit 794db37

Browse files
vinx13masahi
authored andcommitted
Update have_int8 condition to run on compute capability 7.x devices (#4214)
1 parent e889928 commit 794db37

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

python/tvm/contrib/nvcc.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -245,8 +245,8 @@ def have_int8(compute_version):
245245
compute_version : str
246246
compute capability of a GPU (e.g. "6.1")
247247
"""
248-
major, minor = parse_compute_version(compute_version)
249-
if major == 6 and minor == 1:
248+
major, _ = parse_compute_version(compute_version)
249+
if major >= 6:
250250
return True
251251

252252
return False

0 commit comments

Comments
 (0)