use bundled CUDA 10.1 header file
@@ -141,6 +141,8 @@ cpdef function.Module compile_with_cache(
bundled_include = 'cuda-9.2'
elif 10000 <= _cuda_runtime_version < 10010:
bundled_include = 'cuda-10.0'
+ elif 10010 <= _cuda_runtime_version < 10020:
+ bundled_include = 'cuda-10.1'
else:
# CUDA v9.0, v9.1 or versions not yet supported.
bundled_include = None
