Commit 6a98d7f
committed
Fix CuPy JIT: use CUDA devel image for runtime stage
CuPy needs cuda_fp16.h for JIT kernel compilation at runtime.
The runtime CUDA image doesn't include development headers.
Also adds --no-tar flag to pipeline arg parser.1 parent 98277c8 commit 6a98d7f
2 files changed
Lines changed: 4 additions & 1 deletion
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
48 | 48 | | |
49 | 49 | | |
50 | 50 | | |
51 | | - | |
| 51 | + | |
| 52 | + | |
| 53 | + | |
52 | 54 | | |
53 | 55 | | |
54 | 56 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
94 | 94 | | |
95 | 95 | | |
96 | 96 | | |
| 97 | + | |
97 | 98 | | |
98 | 99 | | |
99 | 100 | | |
| |||
0 commit comments