From 8a6f6bc38b17febb15d375fc7de22a4f3e798843 Mon Sep 17 00:00:00 2001 From: Tianlei Wu Date: Mon, 22 Feb 2021 22:00:21 -0800 Subject: [PATCH] add --enable_cuda_line_info to build.py (#6773) --- tools/ci_build/build.py | 3 +++ 1 file changed, 3 insertions(+) mode change 100755 => 100644 tools/ci_build/build.py diff --git a/tools/ci_build/build.py b/tools/ci_build/build.py old mode 100755 new mode 100644 index 521e83c00b..45e1ddef73 --- a/tools/ci_build/build.py +++ b/tools/ci_build/build.py @@ -201,6 +201,8 @@ def parse_arguments(): "--cudnn_home", help="Path to CUDNN home. " "Read from CUDNN_HOME environment variable if --use_cuda is true and " "--cudnn_home is not specified.") + parser.add_argument( + "--enable_cuda_line_info", action='store_true', help="Enable CUDA line info.") # Python bindings parser.add_argument( @@ -715,6 +717,7 @@ def generate_build_tree(cmake_path, source_dir, build_dir, cuda_home, cudnn_home "-DOnnxruntime_GCOV_COVERAGE=" + ("ON" if args.code_coverage else "OFF"), "-Donnxruntime_USE_MPI=" + ("ON" if args.use_mpi else "OFF"), "-Donnxruntime_ENABLE_MEMORY_PROFILE=" + ("ON" if args.enable_memory_profile else "OFF"), + "-Donnxruntime_ENABLE_CUDA_LINE_NUMBER_INFO=" + ("ON" if args.enable_cuda_line_info else "OFF"), ] if acl_home and os.path.exists(acl_home):