Parses command line arguments.
()
| 121 | |
| 122 | |
| 123 | def parse_args(): |
| 124 | """Parses command line arguments.""" |
| 125 | parser = argparse.ArgumentParser() |
| 126 | parser.add_argument( |
| 127 | "--cl_kernel_dir", |
| 128 | type=str, |
| 129 | default="./mace/ops/opencl/cl/", |
| 130 | help="The cl kernels directory.") |
| 131 | parser.add_argument( |
| 132 | "--output_path", |
| 133 | type=str, |
| 134 | default="./mace/examples/codegen/opencl/opencl_encrypted_program.cc", |
| 135 | help="The path of encrypted opencl kernels.") |
| 136 | return parser.parse_known_args() |
| 137 | |
| 138 | |
| 139 | if __name__ == '__main__': |
no outgoing calls
no test coverage detected