From 99d6a4c7266ae39e013a6fea494478628d22258e Mon Sep 17 00:00:00 2001 From: Ben Frederickson Date: Wed, 16 Jun 2021 11:30:24 -0700 Subject: [PATCH] Fix building on CUDA 11.3 --- python/setup.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/setup.py b/python/setup.py index 0346dc713..2663119ca 100644 --- a/python/setup.py +++ b/python/setup.py @@ -78,6 +78,7 @@ def get_cuda_version_from_header(cuda_include_dir): "11.0": "11.x", "11.1": "11.x", "11.2": "11.x", + "11.3": "11.x", } for pxd_basename in files_to_preprocess: