()
| 137 | |
| 138 | |
| 139 | def parse_args(): |
| 140 | parser = argparse.ArgumentParser() |
| 141 | parser.add_argument( |
| 142 | '--binary_files', |
| 143 | type=str, |
| 144 | default="", |
| 145 | help="opencl binary files") |
| 146 | parser.add_argument( |
| 147 | '--tuning_files', |
| 148 | type=str, |
| 149 | default="", |
| 150 | help="tuning params files") |
| 151 | parser.add_argument( |
| 152 | '--output', |
| 153 | type=str, |
| 154 | default="build", |
| 155 | help="output dir") |
| 156 | parser.add_argument( |
| 157 | "--gencode", |
| 158 | action="store_true", |
| 159 | help="generate code") |
| 160 | flgs, _ = parser.parse_known_args() |
| 161 | return flgs |
| 162 | |
| 163 | |
| 164 | if __name__ == '__main__': |