pytorch/caffe2/python/trt
pbialecki 7b85adf20f Add back pycuda.autoinit to test_pt_onnx_trt (#51106)
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
2021-01-27 07:10:11 -08:00
..
data
__init__.py
test_pt_onnx_trt.py Add back pycuda.autoinit to test_pt_onnx_trt (#51106) 2021-01-27 07:10:11 -08:00
test_trt.py
transform.py