Skip to content

Commit d43606b

Browse files
authored
Update torchtrtc.rst
1 parent 5d1acba commit d43606b

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

docsrc/cli/torchtrtc.rst

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -135,12 +135,15 @@ e.g.
135135
136136
torchtrtc tests/modules/ssd_traced.jit.pt ssd_trt.ts "[(1,3,300,300); (1,3,512,512); (1, 3, 1024, 1024)]@f16%contiguous" -p f16
137137
138+
- To include a set of custom operators
138139

139-
To run with custom torch operators
140140
.. code-block:: shell
141-
torchtrtc tests/modules/ssd_traced.jit.pt ssd_trt.ts --custom-torch-ops=<path to custom library> "[(1,3,300,300); (1,3,512,512); (1, 3, 1024, 1024)]@fp16%contiguous" -p f16
142141
142+
torchtrtc tests/modules/ssd_traced.jit.pt ssd_trt.ts --custom-torch-ops=<path to custom library .so file> "[(1,3,300,300); (1,3,512,512); (1, 3, 1024, 1024)]@fp16%contiguous" -p f16
143+
144+
145+
- To include a set of custom converters
143146

144-
To run with custom converters
145147
.. code-block:: shell
146-
torchtrtc tests/modules/ssd_traced.jit.pt ssd_trt.ts --custom-converters=<path to custom library> "[(1,3,300,300); (1,3,512,512); (1, 3, 1024, 1024)]@fp16%contiguous" -p f16
148+
149+
torchtrtc tests/modules/ssd_traced.jit.pt ssd_trt.ts --custom-converters=<path to custom library .so file> "[(1,3,300,300); (1,3,512,512); (1, 3, 1024, 1024)]@fp16%contiguous" -p f16

0 commit comments

Comments
 (0)