From 4eb0783b80a6b95e001b1bb672af3a565d6560c6 Mon Sep 17 00:00:00 2001 From: Yanming Wang Date: Sun, 28 Jun 2020 23:10:19 +0000 Subject: [PATCH] [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 8c3d34af174e..d990a095bc5a 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):