Function build_cpp_examples

scripts/build_cpp_examples.py:23–63

(build_dir: PathLike, trt_dir: PathLike,
                       enable_multi_device: str, loglevel: int)

def build_cpp_examples(build_dir: PathLike, trt_dir: PathLike,
                       enable_multi_device: str, loglevel: int) -> None:
    logging.basicConfig(level=loglevel,
                        format='%(asctime)s - %(levelname)s - %(message)s')
    # Convert input paths to pathlib.Path objects
    build_dir = Path(build_dir)
    trt_dir = Path(trt_dir)

    assert trt_dir.is_dir()

    def cmake_parse(path: PathLike) -> str:
        return str(path).replace("\\", "/")

    # Remove the build directory if it exists
    if build_dir.exists():
        logging.info(f"Removed directory: {build_dir}")
        shutil.rmtree(build_dir)

    # Create the build directory
    build_dir.mkdir(parents=True, exist_ok=True)

    # Change to the build directory
    with working_directory(build_dir):
        # Run CMake with the specified TensorRT directories
        generator = ["-GNinja"] if platform.system() == "Windows" else []
        generate_command = [
            'cmake',
            '-S',
            '..',
            '-B',
            '.',
            f'-DTensorRT_ROOT={cmake_parse(trt_dir)}',
            f'-DENABLE_MULTI_DEVICE={enable_multi_device}',
        ] + generator
        logging.info(f"Executing {generate_command}")
        subprocess.run(generate_command, check=True)

        # Build the project using make
        build_command = ["cmake", "--build", ".", "--config", "Release"]
        logging.info(f"Executing {build_command}")
        subprocess.run(build_command, check=True)


if __name__ == '__main__':