Skip to content

T
tic
Commit 794db370 by Wuwei Lin Committed by masahi
Options

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

parent e8899285
Showing with 2 additions and 2 deletions
python/tvm/contrib/nvcc.py
...@@ -245,8 +245,8 @@ def have_int8(compute_version): ...@@ -245,8 +245,8 @@ def have_int8(compute_version):
compute_version : str compute_version : str
compute capability of a GPU (e.g. "6.1") compute capability of a GPU (e.g. "6.1")
""" """
major, minor = parse_compute_version(compute_version) major, _ = parse_compute_version(compute_version)
if major == 6 and minor == 1: if major >= 6:
return True return True
return False return False
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment