Commit 563e799
committed
Delay importing torch & friends
time python -c 'import cuda.tile'
Before/after: real 0m1.142s / 0m0.054s
Additionally, optimize the steam argument parsing in launch(). Adding a
simple cache shaves off about 50 nanos.
Signed-off-by: Greg Bonik <gbonik@nvidia.com>1 parent de4a85c commit 563e799
4 files changed
Lines changed: 207 additions & 179 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
244 | 244 | | |
245 | 245 | | |
246 | 246 | | |
| 247 | + | |
| 248 | + | |
| 249 | + | |
| 250 | + | |
247 | 251 | | |
248 | 252 | | |
249 | 253 | | |
| |||
0 commit comments