()
| 691 | |
| 692 | |
| 693 | def parse_args(): |
| 694 | # flag definition |
| 695 | from argparse import RawTextHelpFormatter |
| 696 | |
| 697 | parser = argparse.ArgumentParser(formatter_class=RawTextHelpFormatter) |
| 698 | parser.add_argument( |
| 699 | "venv_dir", help="Directory of virtualenv where target tensorflow installed." |
| 700 | ) |
| 701 | parser.add_argument( |
| 702 | "-v", |
| 703 | "--version", |
| 704 | required=False, |
| 705 | default="auto", |
| 706 | help="Version of built packages, defaults to %(default)s. auto: read from VERSION file.", |
| 707 | ) |
| 708 | parser.add_argument( |
| 709 | "--bridge-gcc", |
| 710 | required=False, |
| 711 | choices=VALID_GCC, |
| 712 | help="GCC version to compile tao bridge, required for configure stages.", |
| 713 | ) |
| 714 | parser.add_argument( |
| 715 | "--compiler-gcc", |
| 716 | required=False, |
| 717 | choices=["7.3", "7.5", "default"], |
| 718 | help="GCC version to compile tao compiler, required for configure stages.", |
| 719 | ) |
| 720 | parser.add_argument( |
| 721 | "--skip-tests", |
| 722 | required=False, |
| 723 | action="store_true", |
| 724 | help="Skip all tests.", |
| 725 | ) |
| 726 | parser.add_argument( |
| 727 | "--skip-sanity", |
| 728 | required=False, |
| 729 | action="store_true", |
| 730 | help="Skip all sanity checks.", |
| 731 | ) |
| 732 | parser.add_argument( |
| 733 | "--no-build-for-test", |
| 734 | required=False, |
| 735 | action="store_true", |
| 736 | help="Skip build stages when running test stages. By default, " |
| 737 | "build stages\nwill be triggered before test stages run.", |
| 738 | ) |
| 739 | parser.add_argument( |
| 740 | "--rocm_toolkit_codegen", action="store_true", help="enable codegen using rocm native toolkit." |
| 741 | ) |
| 742 | parser.add_argument( |
| 743 | "-s", |
| 744 | "--stage", |
| 745 | required=False, |
| 746 | choices=[ |
| 747 | "all", |
| 748 | "configure", |
| 749 | "lint", |
| 750 | "build", |
no test coverage detected