From 9d5c50e07b74d7a8078d0e0b50de98f3ec212bdd Mon Sep 17 00:00:00 2001 From: minghaipeng Date: Mon, 24 Mar 2025 08:46:48 +0000 Subject: [PATCH] fix bug when set PADDLE_CUDA_ARCH_LIST but extra_compile_args not given --- python/paddle/utils/cpp_extension/extension_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/utils/cpp_extension/extension_utils.py b/python/paddle/utils/cpp_extension/extension_utils.py index 9ccbec0fc9eb8..04c4470daa8e8 100644 --- a/python/paddle/utils/cpp_extension/extension_utils.py +++ b/python/paddle/utils/cpp_extension/extension_utils.py @@ -560,7 +560,7 @@ def normalize_extension_kwargs(kwargs, use_cuda=False): kwargs['library_dirs'] = library_dirs # append compile flags and check settings of compiler - extra_compile_args = kwargs.get('extra_compile_args', []) + extra_compile_args = kwargs.get('extra_compile_args', {}) if isinstance(extra_compile_args, dict): for compiler in ['cxx', 'nvcc']: if compiler not in extra_compile_args: