()
| 306 | |
| 307 | |
| 308 | def main(): |
| 309 | parser = argparse.ArgumentParser( |
| 310 | description="Kernel Nsys Profiling - Profile custom CUDA C++ " |
| 311 | "kernels with cuda.core" |
| 312 | ) |
| 313 | parser.add_argument( |
| 314 | "-n", |
| 315 | "--array-size", |
| 316 | type=int, |
| 317 | default=50000, |
| 318 | metavar="N", |
| 319 | help="Array size (default: 50,000)", |
| 320 | ) |
| 321 | |
| 322 | args = parser.parse_args() |
| 323 | run(size=args.array_size) |
| 324 | |
| 325 | |
| 326 | if __name__ == "__main__": |
no test coverage detected