| 19 | |
| 20 | |
| 21 | def parse_args(): |
| 22 | parser = argparse.ArgumentParser(description="MeshGEMV on WSE-3") |
| 23 | parser.add_argument("--P", required=True, type=int, help="PEs rectangle size: P x P") |
| 24 | parser.add_argument("--M", required=True, type=int, help="Left vector dimension: 1 x M") |
| 25 | parser.add_argument("--N", required=True, type=int, help="Right matrix dimension: M x N") |
| 26 | parser.add_argument("--group_num", required=True, type=int, help="Reduce group count") |
| 27 | parser.add_argument("--simulator", action="store_true", help="Run the cloud artifact on the simulator") |
| 28 | return parser.parse_args() |
| 29 | |
| 30 | |
| 31 | def main(): |