diff --git a/tests/lit.cfg b/tests/lit.cfg index 5f358b54..1e6e1f1b 100644 --- a/tests/lit.cfg +++ b/tests/lit.cfg @@ -315,80 +315,19 @@ if config.cuda_tensor_root and config.cuda_tensor_root != "OFF": if config.cuda_file_root and config.cuda_file_root != "OFF": clang_arguments += " -I'%s'" -i_subst = 0 -if config.cuda_sdk_root != "OFF": - i_subst = 1 -if config.cuda_dnn_root != "OFF": - i_subst += 2 -if config.cuda_cub_root != "OFF": - i_subst += 4 -if config.cuda_tensor_root != "OFF": - i_subst += 8 -if config.cuda_file_root != "OFF": - i_subst += 16 - -if i_subst == 31: - config.substitutions.append(("%clang_args", clang_arguments % (config.cuda_sdk_root, config.cuda_dnn_root, config.cuda_cub_root, config.cuda_tensor_root, config.cuda_file_root))) -elif i_subst == 30: - config.substitutions.append(("%clang_args", clang_arguments % (config.cuda_dnn_root, config.cuda_cub_root, config.cuda_tensor_root, config.cuda_file_root))) -elif i_subst == 29: - config.substitutions.append(("%clang_args", clang_arguments % (config.cuda_sdk_root, config.cuda_cub_root, config.cuda_tensor_root, config.cuda_file_root))) -elif i_subst == 28: - config.substitutions.append(("%clang_args", clang_arguments % (config.cuda_cub_root, config.cuda_tensor_root, config.cuda_file_root))) -elif i_subst == 27: - config.substitutions.append(("%clang_args", clang_arguments % (config.cuda_sdk_root, config.cuda_dnn_root, config.cuda_tensor_root, config.cuda_file_root))) -elif i_subst == 26: - config.substitutions.append(("%clang_args", clang_arguments % (config.cuda_dnn_root, config.cuda_tensor_root, config.cuda_file_root))) -elif i_subst == 25: - config.substitutions.append(("%clang_args", clang_arguments % (config.cuda_sdk_root, config.cuda_tensor_root, config.cuda_file_root))) -elif i_subst == 24: - config.substitutions.append(("%clang_args", clang_arguments % (config.cuda_tensor_root, config.cuda_file_root))) -elif i_subst == 23: - config.substitutions.append(("%clang_args", clang_arguments % (config.cuda_sdk_root, config.cuda_dnn_root, config.cuda_cub_root, config.cuda_file_root))) -elif i_subst == 22: - config.substitutions.append(("%clang_args", clang_arguments % (config.cuda_dnn_root, config.cuda_cub_root, config.cuda_file_root))) -elif i_subst == 21: - config.substitutions.append(("%clang_args", clang_arguments % (config.cuda_sdk_root, config.cuda_cub_root, config.cuda_file_root))) -elif i_subst == 20: - config.substitutions.append(("%clang_args", clang_arguments % (config.cuda_cub_root, config.cuda_file_root))) -elif i_subst == 19: - config.substitutions.append(("%clang_args", clang_arguments % (config.cuda_sdk_root, config.cuda_dnn_root, config.cuda_file_root))) -elif i_subst == 18: - config.substitutions.append(("%clang_args", clang_arguments % (config.cuda_dnn_root, config.cuda_file_root))) -elif i_subst == 17: - config.substitutions.append(("%clang_args", clang_arguments % (config.cuda_sdk_root, config.cuda_file_root))) -elif i_subst == 16: - config.substitutions.append(("%clang_args", clang_arguments % config.cuda_file_root)) -elif i_subst == 15: - config.substitutions.append(("%clang_args", clang_arguments % (config.cuda_sdk_root, config.cuda_dnn_root, config.cuda_cub_root, config.cuda_tensor_root))) -elif i_subst == 14: - config.substitutions.append(("%clang_args", clang_arguments % (config.cuda_dnn_root, config.cuda_cub_root, config.cuda_tensor_root))) -elif i_subst == 13: - config.substitutions.append(("%clang_args", clang_arguments % (config.cuda_sdk_root, config.cuda_cub_root, config.cuda_tensor_root))) -elif i_subst == 12: - config.substitutions.append(("%clang_args", clang_arguments % (config.cuda_cub_root, config.cuda_tensor_root))) -elif i_subst == 11: - config.substitutions.append(("%clang_args", clang_arguments % (config.cuda_sdk_root, config.cuda_dnn_root, config.cuda_tensor_root))) -elif i_subst == 10: - config.substitutions.append(("%clang_args", clang_arguments % (config.cuda_dnn_root, config.cuda_tensor_root))) -elif i_subst == 9: - config.substitutions.append(("%clang_args", clang_arguments % (config.cuda_sdk_root, config.cuda_tensor_root))) -elif i_subst == 8: - config.substitutions.append(("%clang_args", clang_arguments % config.cuda_tensor_root)) -elif i_subst == 7: - config.substitutions.append(("%clang_args", clang_arguments % (config.cuda_sdk_root, config.cuda_dnn_root, config.cuda_cub_root))) -elif i_subst == 6: - config.substitutions.append(("%clang_args", clang_arguments % (config.cuda_dnn_root, config.cuda_cub_root))) -elif i_subst == 5: - config.substitutions.append(("%clang_args", clang_arguments % (config.cuda_sdk_root, config.cuda_cub_root))) -elif i_subst == 4: - config.substitutions.append(("%clang_args", clang_arguments % config.cuda_cub_root)) -elif i_subst == 3: - config.substitutions.append(("%clang_args", clang_arguments % (config.cuda_sdk_root, config.cuda_dnn_root))) -elif i_subst == 2: - config.substitutions.append(("%clang_args", clang_arguments % config.cuda_dnn_root)) -elif i_subst == 1: - config.substitutions.append(("%clang_args", clang_arguments % config.cuda_sdk_root)) +# Collect all active root paths in the exact order they are appended to clang_arguments +roots = [ + root for root in [ + config.cuda_sdk_root, + config.cuda_dnn_root, + config.cuda_cub_root, + config.cuda_tensor_root, + config.cuda_file_root, + ] if root and root != "OFF" +] + +if roots: + config.substitutions.append(("%clang_args", clang_arguments % tuple(roots))) if config.llvm_version_major < 4: hipify_arguments = "-I'%s'/include"