Commit b25c84a
Fix torch_tensorrt import crash on non-GPU nodes
torch_tensorrt now raises RuntimeError (not ImportError/OSError) when
no CUDA device is available. Add RuntimeError to the except clause so
convnet loads gracefully, allowing mazepa_layer_processing and other
modules to register their builders on non-GPU headnodes.
Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>1 parent e34462a commit b25c84a
1 file changed
Lines changed: 1 addition & 1 deletion
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
24 | 24 | | |
25 | 25 | | |
26 | 26 | | |
27 | | - | |
| 27 | + | |
28 | 28 | | |
29 | 29 | | |
30 | 30 | | |
| |||
0 commit comments