()
| 12 | |
| 13 | |
| 14 | def parse_args(): |
| 15 | parser = argparse.ArgumentParser(description="SUMMA GEMM on simulator") |
| 16 | parser.add_argument("--P", required=True, type=int, help="PEs rectangle size: P x P") |
| 17 | parser.add_argument("--M", required=True, type=int, help="Rows of X") |
| 18 | parser.add_argument("--K", required=True, type=int, help="Inner dimension") |
| 19 | parser.add_argument("--N", required=True, type=int, help="Columns of W") |
| 20 | return parser.parse_args() |
| 21 | |
| 22 | |
| 23 | def main(): |