projects
/
platform
/
upstream
/
tvm.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
| inline |
side by side
(parent:
0465ffd
)
[BUGFIX] Add cuda 11 to contrib.nvcc.find_libdevice_path() (#5902)
author
Yanming Wang
<yanmingwang01@gmail.com>
Sun, 28 Jun 2020 23:10:19 +0000
(23:10 +0000)
committer
GitHub
<noreply@github.com>
Sun, 28 Jun 2020 23:10:19 +0000
(16:10 -0700)
python/tvm/contrib/nvcc.py
patch
|
blob
|
history
diff --git
a/python/tvm/contrib/nvcc.py
b/python/tvm/contrib/nvcc.py
index
8c3d34a
..
d990a09
100644
(file)
--- 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):