From 1c2de952e001337d294be4f79a3ac774191295b5 Mon Sep 17 00:00:00 2001 From: Mancarl Date: Thu, 27 Oct 2022 13:35:24 +0800 Subject: [PATCH] fix error --- kernels/lib_tree_filter/setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/kernels/lib_tree_filter/setup.py b/kernels/lib_tree_filter/setup.py index a4b6398..6ad74ac 100644 --- a/kernels/lib_tree_filter/setup.py +++ b/kernels/lib_tree_filter/setup.py @@ -31,7 +31,7 @@ name='tree_filter_cuda', include_dirs=[extensions_dir], sources=sources, - library_dirs=cuda_lib_path, + library_dirs=include_paths(cuda=True), extra_compile_args={'cxx':['-O3'], 'nvcc':['-O3']}) ],