chore: Review comments incorporated

Signed-off-by: Anurag Dixit <[email protected]>

Author: andi4191

cpp/bin/torchtrtc/README.md

@@ -108,6 +108,8 @@ torchtrtc [input_file_path] [output_file_path]
                                         TorchScript program, save the created
                                         engine to the path specified as the
                                         output path
+      --custom-torch-ops=[lib]          (repeatable) Shared object/DLL containing custom torch operators
+      --custom-converters=[lib]         (repeatable) Shared object/DLL containing custom converters
       input_file_path                   Path to input TorchScript file
       output_file_path                  Path for compiled TorchScript (or
                                         TensorRT engine) file
@@ -131,3 +133,14 @@ e.g.
 ```
 torchtrtc tests/modules/ssd_traced.jit.pt ssd_trt.ts "[(1,3,300,300); (1,3,512,512); (1, 3, 1024, 1024)]@fp16%contiguous" -p f16
 ```
+
+
+To run with custom torch operators
+```
+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
+```
+
+To run with custom converters
+```
+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
+```

cpp/bin/torchtrtc/main.cpp

@@ -176,7 +176,7 @@ int main(int argc, char** argv) {
   args::ValueFlagList<std::string> custom_torch_ops(
       parser,
       "custom-torch-ops",
-      "(repeatable) Shared object/DLL containing custom torch operator",
+      "(repeatable) Shared object/DLL containing custom torch operators",
       {"custom-torch-ops"});
 
   args::ValueFlagList<std::string> custom_converters(

docsrc/tutorials/torchtrtc.rst

@@ -111,6 +111,8 @@ to standard TorchScript. Load with ``torch.jit.load()`` and run like you would r
                                           TorchScript program, save the created
                                           engine to the path specified as the
                                           output path
+        --custom-torch-ops                (repeatable) Shared object/DLL containing custom torch operators
+        --custom-converters               (repeatable) Shared object/DLL containing custom converters
         input_file_path                   Path to input TorchScript file
         output_file_path                  Path for compiled TorchScript (or
                                           TensorRT engine) file
@@ -132,3 +134,13 @@ e.g.
 .. code-block:: shell
 
     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
+
+
+To run with custom torch operators
+.. code-block:: shell
+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
+
+
+To run with custom converters
+.. code-block:: shell
+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