mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
Summary: Fixes https://github.com/pytorch/pytorch/issues/51105 by adding back the `import pycuda.autoinit`. Pull Request resolved: https://github.com/pytorch/pytorch/pull/51106 Reviewed By: mingzhe09088 Differential Revision: D26086808 Pulled By: heitorschueroff fbshipit-source-id: 88d98796c87a44cedaa1f6666e9f71a424293641 |
||
|---|---|---|
| .. | ||
| data | ||
| __init__.py | ||
| test_pt_onnx_trt.py | ||
| test_trt.py | ||
| transform.py | ||