[BUGFIX] Add cuda 11 to contrib.nvcc.find_libdevice_path() (apache#5902)

neo-ai · Jul 2, 2020 · 4eb0783 · 4eb0783
1 parent fcf74a5
commit 4eb0783
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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):