You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by tq...@apache.org on 2020/06/28 23:10:25 UTC
[incubator-tvm] branch master updated: [BUGFIX] Add cuda 11 to
contrib.nvcc.find_libdevice_path() (#5902)
This is an automated email from the ASF dual-hosted git repository.
tqchen pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git
The following commit(s) were added to refs/heads/master by this push:
new 79f3db6 [BUGFIX] Add cuda 11 to contrib.nvcc.find_libdevice_path() (#5902)
79f3db6 is described below
commit 79f3db69c54ef330b311cb31ef4c8796d383553a
Author: Yanming Wang <ya...@gmail.com>
AuthorDate: Sun Jun 28 23:10:19 2020 +0000
[BUGFIX] Add cuda 11 to contrib.nvcc.find_libdevice_path() (#5902)
---
python/tvm/contrib/nvcc.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/python/tvm/contrib/nvcc.py b/python/tvm/contrib/nvcc.py
index 8c3d34a..d990a09 100644
--- a/python/tvm/contrib/nvcc.py
+++ b/python/tvm/contrib/nvcc.py
@@ -173,7 +173,7 @@ def find_libdevice_path(arch):
selected_ver = 0
selected_path = None
cuda_ver = get_cuda_version(cuda_path)
- if cuda_ver in (9.0, 9.1, 10.0):
+ if cuda_ver in (9.0, 9.1, 10.0, 11.0):
path = os.path.join(lib_path, "libdevice.10.bc")
else:
for fn in os.listdir(lib_path):