Parse input arguments.
()
| 25 | |
| 26 | |
| 27 | def parse_args(): |
| 28 | """Parse input arguments.""" |
| 29 | parser = argparse.ArgumentParser(description='Remove grad op and kernels.') |
| 30 | parser.add_argument('--only_kernel', action='store_true', default=False) |
| 31 | parser.add_argument('--dry_run', action='store_true', default=False) |
| 32 | |
| 33 | args = parser.parse_args() |
| 34 | return args |
| 35 | |
| 36 | |
| 37 | def find_type_files(cur_dir, file_type, file_list=[]): |
no test coverage detected