We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6f83b41 commit 7b804a2Copy full SHA for 7b804a2
py/trtorch/_compiler.py
@@ -157,5 +157,6 @@ def get_build_info() -> str:
157
build_info = "TRTorch Version: " + str(__version__) + '\n' + build_info
158
return build_info
159
160
+
161
def set_device(gpu_id):
162
trtorch._C.set_device(gpu_id)
0 commit comments